怎么获得numpy array每行最小值的index?

  统计/机器学习 Python    浏览次数:723        分享
0

如下一个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?

 

小管子   2019-07-27 14:34



   1个回答 
3


>>> 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])


SofaSofa数据科学社区DS面试题库 DS面经

vonta   2019-07-27 16:51

谢谢 - 小管子   2019-07-28 00:19


  相关主题

调用np.average()时报错TypeError: No loop matching   1回答

用np.round的时候报错AttributeError: no attribute 'rint'   1回答

numpy array里怎么用fillna填充nan的值?   1回答

python numpy里array和matrix的区别?   1回答

numpy中获得tensor的维数?   2回答

numpy里面resize和reshape的区别是什么?   1回答

如何往numpy array中插入一列数组?   4回答

怎么对2维的numpy array取整?   2回答

numpy生成二维随机均匀分布?   1回答

怎么把numpy array中的inf换成0?   1回答

怎么把torch.tensor转成numpy ndarray?   2回答

能直接把csv格式的数据文件读到numpy array里吗?   1回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!