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