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

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


  相关讨论

怎么生成一个填满逻辑真(True)的numpy array?

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

numpy里tile函数是怎么用的?

numpy array中元素的个数?

numpy.full这个函数有什么用?

python里sparse matrix怎么转成正常的numpy array显示出来?

怎么随机打乱一个numpy array?

怎么把torch.tensor转成numpy ndarray?

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

numpy中获得tensor的维数?

  随便看看

线性回归或者逻辑回归中常提到的AIC和BIC是什么意思?

两个独立的正态随机变量的乘积服从什么分布?

sklearn中的predict_proba方法的返回值的意义

凸优化中局部最优解就是全局最优解吗?

什么是K-Modes(K众数)聚类法?