python - Pandas :对于一列中的每个项目，计算另一个给定列中不同项目的数量

Question

在给定的 Pandas Dataframe 中，我需要为 x 列中的每个项目计算 y 列中唯一项目的数量。我在字典中想出了计数方法，但它们似乎不能正常工作。我正在尝试管理的示例:

df = pd.DataFrame([[a, 2], [a, 1], [c, 4], [a, 1], [c, 4]]), columns=list('AB'))

创建以下数据框:

   A  B
0  a  2
1  a  1
2  c  4
3  a  1
4  c  4

现在我想要的输出应该是:

a:2
c:1

Answer 1

你需要 groupby 和 SeriesGroupBy.nunique最后to_dict :

print (df.groupby('A').B.nunique())
A
a    2
c    1
Name: B, dtype: int64

print (df.groupby('A').B.nunique().to_dict())
{'a': 2, 'c': 1}