欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Numpy中argsort()、bincount()、argmax()函数用法

程序员文章站 2022-04-01 10:44:12
...

 

np.argsort():可以对数组排序并返回排序后数据的下标,默认为从小到达的排序

>>> import numpy as np
>>> a = [1,2,3,4,2,3,1]
>>> a = np.array(a)
>>> order = np.argsort(a)
>>> order

结果 

array([0, 6, 1, 4, 2, 5, 3])

再根据下标可得到排序后的数组

>>> a[order]
array([1, 1, 2, 2, 3, 3, 4])

np.bincount():会统计一个非负数组中各数据出现的频次

>>> a =[1,2,3,4,2,3,1]
>>> a = np.array(a)
>>> np.bincount(a)
array([0, 2, 2, 2, 1])

从0开始统计到数组中最大值出现的次数,上述数组中4最大,即返回0-4出现的次数

0出现0次,1出现2次,2出现2次,3出现2次,4出现1次

np.argmax():

 

np.random.choice()

 

相关标签: numpy