np.bincount方法

合集 - python(1)

1.np.bincount方法2023-05-30

收起

官方文档
out = np.bincount(x[, weights, minlength])
该函数用于统计输入数组内每个数值出现的次数，输出数组中的索引值对应的是输入数组中的元素值，若输入数组中的某个数值出现了一次，则输出数组对应索引值上的数加一

某个数值n在输入数组x中每出现1次，则输出o内的o[n]+=1

参数

• x: 输入，1维非负数组
• weights: 权重数组, 可选参数，如果指定了这一参数， 则某个数值n在输入数组x中每出现1次，假设这个数在x中的索引值是i， 则输出o内的o[n]+=weights[i]
• minlength: 输出数组最短长度，可选参数。若指定了这个值，则当输出长度不足minlength时，会自动用0补齐，保证输出长度不小于minlength。

示例

a = [1, 2, 2, 3, 2, 3]
b = np.bincount(a)
print(b)
# b = [0, 1, 3, 2]
c = np.bincount(a, minilength=6)
print(c)
# c = [0, 1, 3, 2, 0, 0]
w = [0.1, 0.2, 0.3, 0.2, 0.2, 0.4]
c = np.bincount(a, weights=w)
print(c)
# c = [0, 0.1, 0.7, 0.6]