Python 列表到 np.array 的计数答案

【问题标题】：Python list to np.array of countsPython 列表到 np.array 的计数
【发布时间】：2020-12-08 13:49:19
【问题描述】：

假设我们有一个向量大小N=1000，假设我们得到了列表[1,1,2,2,2,100]

我想生成一个大小为 1000 的 np.array（或 pd.Series），其中 v[n] 是 n 在列表中出现的次数。在我们的示例中，v[1] = 2, v[2] = 3, v[100] = 1, v=[42] = 0

如何优雅地使用 numpy/pandas 做到这一点？

【问题讨论】：

【解决方案1】：

如果你有一个列表mylist，你可以得到一个计数数组mycount：

N = 1000
x = np.array(mylist)
mycount = np.bincount(x, minlength=N)

这会将数组中的每个元素根据其值和数量分类到 bin 中。你可以在this doc page找到更多关于bincount的信息。

【讨论】：

【解决方案2】：

Python 有一个用于计算出现次数的本机方法，称为 Counter，如果需要，可以在不调用 numpy 或 pandas 的情况下使用它

from collections import Counter
a = [1,1,2,2,2,100]
cnts = Counter(a)
print(cnts)
# Counter({2: 3, 1: 2, 100: 1})

您可以将其转换为带有列表理解的列表：

N = 100
cnts_list = [cnts.get(i, 0) for i in range(N+1)]

【讨论】：

【解决方案3】：

使用Series.value_counts 和Series.reindex 添加不存在的值：

a = [1,1,2,2,2,100]

N = 100
a = pd.Series(a).value_counts().reindex(range(N+1), fill_value=0)
print (a)
0      0
1      2
2      3
3      0
4      0
      ..
96     0
97     0
98     0
99     0
100    1
Length: 101, dtype: int64

【讨论】：

【解决方案4】：

您也可以使用np.unique。

N = 1000
result = np.zeros(N)
idx, val = np.unique([1,1,2,2,2,100], return_counts=True)
result[idx] = val
print(result[:5])                                                                                                                                                                                                                                                           
>>>[0. 2. 3. 0. 0.]

【讨论】：

【解决方案5】：

您可以使用系列和分组方式

In[1]:

import pandas as pd
my_list = [1,1,1,2,2,2,2,3,4,8,1000,8,8,5,5,6]

my_Serie = pd.Series(my_list)
v = my_Serie.groupby(my_list).count().to_dict()
print(v)

{1: 3, 2: 4, 3: 1, 4: 1, 5: 2, 6: 1, 8: 3, 1000: 1}

【讨论】：