"['カラム名'] not found in axis"とエラーが出る。

Question

前提・実現したいこと

kaggleでしたのコードを実装したところ、"['game_session'] not found in axis"とエラーが出ましたが、なぜでしょうか？

group_game_time = df.drop(columns=['event_id', 'event_code']).groupby(
['game_session', 'installation_id', 'title', 'type', 'world']
).max()

group_game_time.drop(columns=['game_session', 'event_count', 'game_time'])

reset_index()を追加したところ解決できましたが、理由は不明です。

group_game_time = df.drop(columns=['event_id', 'event_code']).groupby(
['game_session', 'installation_id', 'title', 'type', 'world']
).max().reset_index()

group_game_time.drop(columns=['game_session', 'event_count', 'game_time'])

Answer 1

groupby をすると、そこで使ったカラムがindexになってしまうため、カラムとしての指定ができなくなってしまいます。
reset_index をすると動くようになるのは、groupby でindexになったカラムが、元のようにカラムとして扱われるからです。

別のやり方として、groupbyする際に、オプションで as_index=False と指定して、そもそもカラムをindexにしないことも可能です。

group_game_time = df.drop(columns=['event_id', 'event_code']).groupby(
['game_session', 'installation_id', 'title', 'type', 'world'], as_index=False).max()

Comments

[gjgjgjgjg]

ありがとうございます！
理解できました、
見返す時の為に参考したリンクも載せておきます。

group byの際にindexになる記事
https://qiita.com/propella/items/a9a32b878c77222630ae

reset_indexでindexに名前がついていた場合に、再度その名前を利用してカラムを作成する記事
https://note.nkmk.me/python-pandas-reset-index/