web-dev-qa-db-ja.com

(行、列)形式のNumpy最小値

どうすればnumpy配列/行列の最小値の(行、列)インデックスを知ることができますか?

たとえば、A = array([[1, 2], [3, 0]])の場合、(1, 1)を取得します

ありがとう!

46
yassin

使用する - unravel_index

numpy.unravel_index(A.argmin(), A.shape)
76
Philipp

[誤植を訂正]

別の簡単な解決策は

ri, ci = A.argmin()//A.shape[1], A.argmin()%A.shape[1]

Numpy.argminは行優先順でインデックスの読み取りを返すため


はい、そうです、それは正方行列で、正方行列で機能しました

2
Nikesh Bajaj