pd.DataFrameから、ある列で同じ値のものを分割して別々のDataFrameにしたい

Question

###前提・実現したいこと
id スコア
0001 1528
0001 105090
0001 78936806
0002 23
0002 54170
0003 6952
0003 748569120
0004 14680
0004 69325
...
...
というようなcsvから読み込んだDataFrameを
同じidごとに分類してDataFrameを分けたいのですがうまくいきません。
idは数十万あります。

どのように書いたら良いでしょうか？

Answer 1

DataFrame.groupby() をお使いください。

# idでグループ化
grouped_df = df.groupby('id')
# グループ一覧
print(grouped_df.groups)
# グループ（id='0001'）のDataFrameを取得
print(grouped_df.get_group('0001'))

ですので、全てのグループに処理を行うにはループをつかって

#グループ（id）毎のDataFrameを順番に取得
for id in grouped_df.groups:
   d = grouped_df.get_group(id)

のように記述できるかと思います。
また、内包表記により

grouped_list = [grouped_df.get_group(id) for id in grouped_df.groups]

のようにリスト化することもできます。

Comments

[machiko4740]

ありがとうございます????
やってみます！