标签：argmin argmax print np import array numpy axis

numpy.argsort(), numpy.argmax(), numpy.argmin()用法

• numpy.argsort(a, axis=-1, kind=’quicksort’, order=None)

功能：将矩阵a按照axis排序，并返回排序后的索引

参数：a为输入矩阵，axis为需要排序的维度，axis=0按列排序，axis=1按行排序

返回值：排序后的索引

# 一维向量
import numpy as np
a = np.array([1, 2, 3])
b = np.argsort(a)
print(b)
>> Out: [0 2 1]

# 二维向量，axis为默认值
import numpy as np
a = np.array([[1, 3, 2],[5, 7, 6]])
b = np.argsort(a)
print(b)
>> Out: [[0 2 1]
         [0 2 1]]

# 二维向量，axis为0
import numpy as np
a = np.array([[1, 3, 2],[5, 7, 6]])
b = np.argsort(a, axis=0)
print(b)
>> Out: [[0 0 0]
         [1 1 1]]

• numpy.max(a, axis=-1, kind=’quicksort’, order=None)

功能：找到指定axis最大值，并返回最大值的索引

参数：a为输入矩阵，axis为寻找最大值的维度，axis=0按列寻找，axis=1按行寻找

返回值：最大值的索引

# 一维向量
import numpy as np
a = np.array([1, 2, 3])
b = np.argmax(a)
print(b)
>> Out: 2

# 二维向量，axis为默认值
import numpy as np
a = np.array([[1, 3, 2],[5, 7, 6]])
b = np.argmax(a)
print(b)
>> Out: 4

# 二维向量，axis为0
import numpy as np
a = np.array([[1, 3, 2],[5, 7, 6]])
b = np.argmax(a, axis=0)
print(b)
>> Out: [1 1 1]
    
# 二维向量，axis为1
import numpy as np
a = np.array([[1, 3, 2],[5, 7, 6]])
b = np.argmax(a, axis=1)
print(b)
>> Out: [1 1]
    
# 三维向量，axis为默认值
import numpy as np
a = np.array([[[1, 3, 2],[5, 7, 6]], [[4, 8, 6],[5, 7, 9]]])
b = np.argmax(a)
print(b)
>> Out: 11
    
# 三维向量，axis为0
import numpy as np
a = np.array([[[1, 3, 2],[5, 7, 6]], [[4, 8, 6],[5, 7, 9]]])
b = np.argmax(a, axis=0)
print(b)
>> Out: [[1 1 1]
         [0 0 1]]

# 三维向量，axis为1
import numpy as np
a = np.array([[[1, 3, 2],[5, 7, 6]], [[4, 8, 6],[5, 7, 9]]])
b = np.argmax(a, axis=1)
print(b)
>> Out: [[1 1 1]
         [1 0 1]]

• numpy.min(a, axis=-1, kind=’quicksort’, order=None)

功能：找到指定axis最小值，并返回最小值的索引

参数：a为输入矩阵，axis为寻找最小值的维度，axis=0按列寻找，axis=1按行寻找

返回值：最小值的索引

标签：argmin,argmax,print,np,import,array,numpy,axis
来源： https://blog.csdn.net/qq_38392644/article/details/104834025

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。