Pandas DataFrame の groupbyについて

pandas DataFrame のgroupby メソッドの理解が不十分だと感じていたのでメモすることにした。 まずは、題材となるデータフレームとその処理コードを書く。

import pandas as pd, numpy as np
df = pd.DataFrame({"a":["A","A","B","B","C"],"b":[1,2,3,3,2],"c":[np.nan,np.nan,2,5,2]})
gb = df.groupby(['a'],as_index=False)[["b","c"]].count()

とすると、df,gbの出力はそれぞれ図１、図２になる。

<ポイント１>as_index=False このオプションをつけることでキーがインデックスではなくカラムに出力される。 個人的にはこれをデフォルトにすればいいのにと思う。

<ポイント2> countは、nanとなっているものについてはカウントしない。

図１：dfの出力。b列,c列の違いはnanになっているかどうか。

図２：gbの出力。キーであるaがインデックスではなくカラムにあることと、b列,c列で出力に違いがあることに注意。