计算熊猫中的项目数

Question

我知道这个问题很琐碎，但这是我使用 pandas 的第一天，所以请至少发布一个指向我应该阅读的文档的链接。

基本上，我不知道如何计算每列的项目数。假设我有一个这样的数据框：

df = pandas.DataFrame({
    "Grade": [1, 2, 1, 1, 2, 1, 2, 2, 1],
    "Major": ["Science", "Art", "Engineering", "Science", "Science", "Science", "Science", "Engineering", "Engineering"]})

我希望得到以下结果：

Grade   Major           Number
1       Science         3
1       Engineering     2
1       Art             0
2       Science         2
2       Engineering     1
2       Art             1

Answer 1

我认为您需要 groupby 和聚合 size：

print (df.groupby(['Grade','Major']).size().reset_index(name='Number'))
   Grade        Major  Number
0      1  Engineering       2
1      1      Science       3
2      2          Art       1
3      2  Engineering       1
4      2      Science       2

如果需要为每个组添加缺失值，请添加unstack 和stack：

print (df.groupby(['Grade','Major'])
         .size()
         .unstack(fill_value=0)
         .stack()
         .reset_index(name='Number'))
   Grade        Major  Number
0      1          Art       0
1      1  Engineering       2
2      1      Science       3
3      2          Art       1
4      2  Engineering       1
5      2      Science       2