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

  统计/机器学习 Python    浏览次数:357        分享
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


  相关主题

numpy里的np.newaxis有什么用?   1回答

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

怎么用numpy生成一个三维的tensor?   1回答

求一个nxn的numpy array的对角线的和?   1回答

'numpy.ndarray' object has no attribute 'head'   1回答

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

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

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

删除numpy array中指定的一列   1回答

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

对numpy array求每行的均值   1回答

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



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

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

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

我们谢绝答非所问。

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

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