np.bincount()
时间: 2024-12-17 09:34:35 浏览: 5
numpy 计算两个数组重复程度的方法
`np.bincount()`是NumPy库中的一个函数,主要用于统计离散整数数组中各个元素的频数。它的基本用法是对输入数组中的每个元素,计算该元素在指定范围内出现的次数,并返回一个新的计数数组。这个函数常用于频率分析、离散数据的可视化以及生成直方图的基础。
例如,如果你有一个表示班级成绩的整数数组,你可以使用`bincount`来获取每个分数段的学生人数:
```python
import numpy as np
scores = [85, 90, 75, 85, 92, 78]
counts = np.bincount(scores)
```
这将返回一个一维数组 `[3, 2, 1]`,分别对应85分、90分和75分的学生人数。
`np.bincount()`还可以接受额外的两个参数,` minlength` 和 `weights`,用来进一步定制功能:
- `minlength` 指定输出数组的长度,如果不设置则会自动填充最大值加1。
- `weights` 可以是一个数组,其中每个元素对应输入数组的一个元素,`weights[i]` 将乘以 `i` 后作为 `np.bincount()` 的输入。
使用`weights`时,`bincount`实际上是计算每个值在给定权重下的总和,而非简单计数。
需要注意的是,`np.bincount()`对于负数或浮点数不会直接工作,通常需要先转换成整数范围内的索引。
阅读全文