NumPy – 统计与排序

统计与排序


常用函数

函数 作用
np.argmax() 返回最大值的下标
np.argsort() 排序后的下标(从小到大)
np.sum() 求和
np.bincount() 返回一组序列,分别记录对应下标值在原序列中出现的次数

np.arg系列函数

  np.arg系列函数返回某种计算结果的下标,包含np.argmax,np.argsort,np.argwhere等函数。arg系函数都有一个axis参数,表示在哪一维上进行计算。举个栗子,用np.argmax函数计算3×3矩阵的最大值下标,不同的axis取值计算过程如下:

axis 计算过程 结果
不指定 将矩阵展平后计算 最大值下标(0-8之间的一个数)
0 第0维上找最大值下标 argmax([0:3,0]),argmax([0:3,1]),argmax([0:3,2])
1 第1维上找最大值下标 argmax([0,0:3]),argmax([0,0:3]),argmax([0,0:3])

np.bincount

  对于输入的整数序列,返回一组序列,记录每个下标值出现的次数,返回序列的最大下标值为原序列的最大元素值。如果输入的系列是浮点数序列,则先进行取整处理。

发表评论

电子邮件地址不会被公开。 必填项已用*标注