首页 > 编程语言 > 详细

Python3 求最大/小值及索引值 Numpy

时间:2018-11-19 18:31:47      阅读:290      评论:0      收藏:0      [点我收藏+]

Python3 求最值并返回索引值 Numpy

np.argmax()

mp.argmin()

import numpy as np

c = np.array([[11, 2, 8, 4], [4, 52, 6, 17], [2, 8, 9, 100]])
print(c)

print(np.argmin(c))
print(np.argmin(c, axis=0)) # 按每列求出最小值的索引
print(np.argmin(c, axis=1)) # 按每行

输出结果:

[[ 11 2 8 4]
[ 4 52 6 17]
[ 2 8 9 100]]
1
[2 0 1 0]
[1 0 0]

np.argwhere()

>>> x = np.arange(6).reshape(2,3)
>>> x
array([[0, 1, 2],
       [3, 4, 5]])
>>> np.argwhere(x>1)
array([[0, 2],
       [1, 0],
       [1, 1],
       [1, 2]])

np.argsort()

One dimensional array:

>>> x = np.array([3, 1, 2])
>>> np.argsort(x)
array([1, 2, 0])

Two-dimensional array:

>>> x = np.array([[0, 3], [2, 2]])
>>> x
array([[0, 3],
       [2, 2]])
>>> np.argsort(x, axis=0)
array([[0, 1],
       [1, 0]])
>>> np.argsort(x, axis=1)
array([[0, 1],
       [0, 1]])

Sorting with keys:

>>> x = np.array([(1, 0), (0, 1)], dtype=[(‘x‘, ‘<i4‘), (‘y‘, ‘<i4‘)])
>>> x
array([(1, 0), (0, 1)],
      dtype=[(‘x‘, ‘<i4‘), (‘y‘, ‘<i4‘)])
>>> np.argsort(x, order=(‘x‘,‘y‘))
array([1, 0])
>>> np.argsort(x, order=(‘y‘,‘x‘))
array([0, 1])

 

 

 

 

Python3 求最大/小值及索引值 Numpy

原文:https://www.cnblogs.com/yuanji2018/p/9984492.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!