矢量化實現，以獲取每行中最小元素的索引

Question

我有一個像這樣的numpy.ndarray：

array([[ 11.18033989,   0.        ],
       [  8.24621125,   3.        ],
       [ 13.03840481,   5.        ],
       [  6.        ,   5.38516481],
       [ 11.18033989,   3.16227766],
       [  0.        ,  11.18033989],
       [  8.06225775,   4.24264069]])

我想得到一個新的數組A，這樣A [i]是上面矩陣的第i行中最小元素的索引。 如： array([1, 1, 1, 1, 1, 0, 1])

我可以使用argmin for for循環，但由於我希望這個算法可擴展，我正在尋找一種使用矢量化實現的方法。 我猜numpy會提供這樣的功能，但我是numpy的新手，所以我不知道在哪里看。

Answer 1

如果X是你的數組，

X.argmin(axis=1)