どうすればnumpy配列/行列の最小値の(行、列)インデックスを知ることができますか?
たとえば、A = array([[1, 2], [3, 0]])
の場合、(1, 1)
を取得します
ありがとう!
使用する - unravel_index
:
numpy.unravel_index(A.argmin(), A.shape)
[誤植を訂正]
別の簡単な解決策は
ri, ci = A.argmin()//A.shape[1], A.argmin()%A.shape[1]
Numpy.argminは行優先順でインデックスの読み取りを返すため
はい、そうです、それは正方行列で、正方行列で機能しました