如下一个array
[[0, 1, 5],
[2, 3, 1],
[1, 4, 2]]第一行最小的是0，index是0；第二行最小的是1，index是2；第三行最小的是1，index=0。
上面这个array每行最小值分别为[0, 2, 0]。
请问有什么方法可以像如上获取numpy array每行最小值的index？
1个回答
>>> a = np.arange(6).reshape(2,3) + 10
>>> a
array([[10, 11, 12],
[13, 14, 15]])
>>> np.argmin(a)
0
>>> np.argmin(a, axis=0)
array([0, 0, 0])
>>> np.argmin(a, axis=1)
array([0, 0])
vonta
2019-07-27 16:51
谢谢
-
小管子
2019-07-28 00:19
相关讨论
随便看看