矢量化实现，以获取每行中最小元素的索引

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)