如下一个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
相关主题
调用np.average()时报错TypeError: No loop matching
1回答
能直接把csv格式的数据文件读到numpy array里吗？
1回答
用np.round的时候报错AttributeError: no attribute 'rint'
1回答
怎么生成一个填满逻辑真（True）的numpy array？
1回答
我们谢绝在回答前讲“生动”的故事。
我们谢绝“这么简单，你自己想”、“书上有的，你认真看”这类的回答；如果你认为对方的提问方式或者内容不妥，你可以直接忽略该问题，不用进行任何作答，甚至可以对该问题投反对票。
我们谢绝答非所问。
我们谢绝自己不会、硬要回答。
我们感激每一个用户在编写答案时的努力与付出！