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

我有这样一个小房间：

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行中最小元素的索引。例如：

数组（[1,1,1,1,1,0,1]）

---

我可以用argmin实现for循环，但是因为我希望这个算法是可伸缩的，所以我正在寻找一种使用矢量化实现的方法。我想numpy会提供这样的功能，但我对numpy还不熟悉，所以我不确定该去哪里找。

如果

X

是您的阵列

X.argmin(axis=1)

---

我原以为这是件容易的事，但这超出了我的预料。谢谢。