एक धुरी

Question

के साथ एक numpy सरणी में अधिकतम तत्व की अनुक्रमणिका कैसे प्राप्त करें मेरे पास 2 आयामी NumPy सरणी है। मुझे पता है कि axes पर अधिकतम मान कैसे प्राप्त करें:

>>> a = array([[1,2,3],[4,3,1]]) 
>>> amax(a,axis=0) 
array([4, 3, 3]) 

मैं अधिकतम तत्वों के सूचकांक कैसे प्राप्त कर सकता हूं? तो मैं आउटपुट के रूप में चाहते हैं array([1,1,0])

Answer 1

>>> a.argmax(axis=0) 

array([1, 1, 0]) 

Answer 2

v = alli.max() 
index = alli.argmax() 
x, y = index/8, index%8 

Answer 3

>>> import numpy as np 
>>> a = np.array([[1,2,3],[4,3,1]]) 
>>> i,j = np.unravel_index(a.argmax(), a.shape) 
>>> a[i,j] 
4 

Answer 4

argmax() केवल प्रत्येक पंक्ति के लिए पहली घटना वापस आ जाएगी। http://docs.scipy.org/doc/numpy/reference/generated/numpy.argmax.html

क्या तुमने कभी एक आकार सरणी के लिए यह करने के लिए की जरूरत है, इस काम करता है unravel की तुलना में बेहतर:

import numpy as np 
a = np.array([[1,2,3], [4,3,1]]) # Can be of any shape 
indices = np.where(a == a.max()) 

आप भी अपनी स्थिति बदल सकते हैं:

indices = np.where(a >= 1.5) 

ऊपर आप परिणाम देता है जिस रूप में आपने पूछा था। वैकल्पिक रूप से, आप एक्स, वाई निर्देशांक की एक सूची में परिवर्तित कर सकते हैं:

x_y_coords = zip(indices[0], indices[1])