pandasでdf.groupby('key1')[['data2']]のようにネストされたカラム名の意味を教えて下さい

Question

オライリーのPythonによるデータ分析入門第２版

df = pd.DataFrame({'key1': ['a', 'a', 'b', 'b', 'a'],
                  'key2': ['one', 'two', 'three', 'one', 'two'],
                  'data1': np.random.randn(5),
                  'data2': np.random.randn(5)})
df.groupby('key1')[['data2']]
#df[['data2']].groupby(df['key1'])のシンタックスシュガー

p321にdf.groupby('key1')[['data2']]のような表記があります。なぜネストしているのか分かりません。この意味を教えて頂けないでしょうか？
また、df.groupby('key1')['data2']との違いは何なのでしょうか？
宜しくお願いします。

Answer 1

pandasで、df[a_column]とdf[[a_column]]の違いは、SeriesかDataFrameかの違いです。

今回はGroupByオジェクトなので、以下のように違います。

>>> print(type(df.groupby('key1')[['data2']]))
<class 'pandas.core.groupby.generic.DataFrameGroupBy'>
>>> print(type(df.groupby('key1')['data2']))
<class 'pandas.core.groupby.generic.SeriesGroupBy'>

Comments

[ao_tombo]

ご回答ありがとうございました。良く分かりました！