python - Pandas 相当于 SQL where

Question

我是 Pandas 的新手，遇到了一些麻烦。基本上我正在尝试实现 SQL 查询

select count(fraud),state
from table
where fraud='REJECT'
group by state

我有以下python代码

df.groupby('State').size()

然而，这并不仅限于 fraud=='REJECT'。我试过了

fraud=df['fraud']=='REJECT'
fraud.groupby('State').size()

然而，这会为“State”创建一个关键错误。所以我认为这归结为我不知道如何在 Pandas 中实现 SQL“where”。有人可以帮我吗？谢谢

Answer 1

你可以这样做:

df[df['fraud'] == 'REJECT'].groupby('State').size()

例子:

>>> df = pd.DataFrame({'fraud':['REJECT', 'ACCEPT', 'REJECT', 'REJECT'], 'State':['AZ', 'AZ', 'TX', 'TX']})
>>> df[df['fraud'] == 'REJECT'].groupby('State').size()
State
AZ       1
TX       2
dtype: int64