Sayısal dizide en yüksek değeri içeren satır veya sütunu bulma

Question

2d numpy dizisinde dizi boyu maksimum değerini içeren satırı veya sütunu nasıl bulabilirim?

Answer 1

gibi np.unravel_index birlikte np.argmax kullanabilirsiniz: sütun için

np.argmax(np.max(x, axis=1)) 

ve için

np.argmax(np.max(x, axis=0)) 

kürek çekmek.

Answer 2

np.where(x == np.max(x))'u kullanabilirsiniz. Örneğin

:

>>> x = np.array([[1,2,3],[2,3,4],[1,3,1]]) 
>>> x 
array([[1, 2, 3], 
     [2, 3, 4], 
     [1, 3, 1]]) 
>>> np.where(x == np.max(x)) 
(array([1]), array([2])) 

ilk değer, satır sayısı, ikinci sayı sütun sayısıdır.

Answer 3

yalnızca birini ya da diğer ihtiyaç Sen

x = np.random.random((5,5)) 
print np.unravel_index(np.argmax(x), x.shape) 

Answer 4

np.argmax yalnızca düzleştirilmiş dizideki (ilk) en büyük öğenin dizinini döndürür. Yani dizininizin şeklini biliyorsanız (hangisini yaparsanız) satır/sütun dizinlerini kolayca bulabilirsiniz:

A = np.array([5, 6, 1], [2, 0, 8], [4, 9, 3]) 
am = A.argmax() 
c_idx = am % A.shape[1] 
r_idx = am // A.shape[1]