60

numpy配列/行列の最小値の(行、列)インデックスを知るにはどうすればよいですか?

たとえばA = array([[1, 2], [3, 0]])(1, 1)

ありがとう!

4

2 に答える 2

109

使用unravel_index

numpy.unravel_index(A.argmin(), A.shape)
于 2010-07-12T16:13:13.420 に答える
3

[タイプミスを修正]

別の簡単な解決策は

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

numpy.argminは、行の主要な順序でインデックスの読み取り値を返すため


はい、あなたは正しいです、それは正方行列のために働いたタイプミスでした

于 2018-03-27T08:25:08.480 に答える