温馨提示×

NumPy怎么统计元素出现次数

小亿
323
2024-05-13 12:41:12
栏目: 编程语言

要统计NumPy数组中每个元素出现的次数,可以使用numpy.unique()函数来获取数组中唯一元素的频率,然后使用numpy.bincount()函数来统计每个元素出现的次数。下面是一个示例:

import numpy as np

# 创建一个NumPy数组
arr = np.array([1, 2, 2, 3, 3, 3, 4, 4, 4, 4])

# 获取数组中每个元素的唯一值和它们的频率
unique_values, counts = np.unique(arr, return_counts=True)

# 创建一个长度为数组最大值加1的数组,用于统计每个元素的出现次数
freq = np.zeros(np.max(arr)+1)
freq[unique_values] = counts

print(freq)

运行以上代码后,将输出以下结果,其中数组中的每个元素表示对应元素的出现次数:

[0. 1. 2. 3. 4.]

0