python - 仅显示一个类别的总和函数值

Question

data = {'Person': ['a','b','c','d','a','b','c','d','b','c'],
        'months':['Jan','Jan','Jan','Jan','Feb','Feb','Feb','Feb','March','March'],
        'income':[100,75,80,56,48,56,37,48,95,65]}
df = pd.DataFrame(data)

df.groupby(['Person'])['income'].sum()

输出：

Person  
a    148  
b    226  
c    182  
d    104  
Name: income, dtype: int64

但我只想显示数据a。我怎样才能做到这一点？

标签： pythonpandasdataframe

Answer 1

df[df['Person'] == 'a'].groupby(['Person'])['income'].sum()

输出

Person
a    148
Name: income, dtype: int64