なぜpandasはgroupby文の行を削除して表示しますか？

Question

私は、単純なパンダのデータフレームを持っている：

 A  B 
0 test fast 
1 train slow 
2 test fast 
3 train slow 

今私は、カラムA ==テスト

df2.drop(df2[df2['A'] == 'test'].index, inplace=True) 

行を削除し、結果として得る：

 A  B 
1 train slow 
3 train slow 

今私が実行groupbyの声明：

df2.groupby('A').B.count() 

結果として得る：それは削除されたものの

A 
test  0 
train 2 

は、なぜ私はテストのための出力を得るのですか？ これを避ける方法は？

あなたのケースのおかげで、 サイモン

Answer 1

は、Aのデータ型がcategoryで、以下を参照：

をカテゴリに変換Aの後に、私はあなた

df.A=df.A.astype('category') 
df1=df.drop(df[df['A'] == 'test'].index) 
df1.groupby('A').B.count() 

A 
test  0 
train 2 
Name: B, dtype: int64 

と同じ結果を得ましたあなたが望む出力を得るには、元のdf.Aをstringに変換してください：

df.A=df.A.astype('str') 
df1=df.drop(df[df['A'] == 'test'].index) 
df1.groupby('A').B.count() 

Out[201]: 
A 
train 2 
Name: B, dtype: int64