首页 > 解决方案 > Numpy计算数组中的项目并存储在字典中

问题描述

可以说我有多个这样的numpy数组:[1, 2, 4, 7, 1, 4, 6, 8, 1, 8, 2, 5]

我想计算每个项目出现在数组中的次数,并将结果存储在字典中:

{1: 3, 2: 2, 3: 0, 4: 2, 5: 1, 6: 1, 7: 1, 8: 2}

有没有比简单地遍历数组并计算项目并将它们存储在字典中更快的方法来做到这一点?

标签: pythonarraysnumpydictionarycount

解决方案


Numpy 就其智慧而言,具有以下功能:

np.unique(x, return_counts=True)

它不返回字典,但您可以轻松地将结果转换为字典。


推荐阅读