DataFrameをグループ化して文字列を結合したい

Question

前提・実現したいこと

PythonのPandasでデータを読み込み、特定の列の値でグループ化した上で、
別の列の文字列をカンマで区切って結合させたいと考えています。

具体的には、伝票番号ごとに単品の注文内容が列挙されているデータを、
伝票単位で注文内容を（カンマで区切って）まとめたいと考えています。

該当のソースコード

import pandas as pd
order_df = pd.DataFrame({'伝票':['1', '1', '2', '2', '1'],
                         '注文':['ビール', '枝豆', 'ウイスキー', 'チョコ', '刺身']})

期待する結果

after_df = pd.DataFrame({'伝票':['1','2'],
                         '注文':['ビール,枝豆,刺身','ウイスキー,チョコ']})

試したこと

下記の方法では、「ビール枝豆刺身」のように、区切りがなく文字列がそのまま結合されてしまいます。

after_df = order_df.groupby('伝票').apply(lambda x: x.sum()).drop('伝票',axis=1).reset_index()

（カンマで区切るのではなく）リスト化させる方法として、下記の方法も見受けられたのですが、
「TypeError: 'DataFrameGroupBy' object is not callable」というエラーが出てしまいました。

after_df = order_df.groupby("伝票")["注文"].apply(list)

Answer 1

シンプルに実装するとこんな感じになります。

df = pd.DataFrame({'伝票':['1', '1', '2', '2', '1'],
                   '注文':['ビール', '枝豆', 'ウイスキー', 'チョコ', '刺身']})


result = df.groupby('伝票')['注文'].apply(','.join).reset_index()
#  伝票         注文
#0  1  ビール,枝豆,刺身
#1  2  ウイスキー,チョコ

Comments

[frederick_1974]

ありがとうございます。
シンプルすぎてびっくりしたのですが、これで実現できるのですね。

Answer 2

次のようにすると期待する結果になるかと思います。

after_df = order_df.groupby('伝票').apply(lambda x: pd.Series({'注文': ','.join(x['注文'])})).reset_index()

Comments

[frederick_1974]

ありがとうございます。
実現できました。Seriesとしてjoinする方法があるのですね。

Answer 3

pandasの上でやろうとするとかえって厄介な気がしたので、groupbyしたあとさっさとイテレータにして内包表記で整えてから再度DataFrameに変換しました。

import pandas as pd

order_df = pd.DataFrame({'伝票':['1', '1', '2', '2', '1'],
                         '注文':['ビール', '枝豆', 'ウイスキー', 'チョコ', '刺身']})
result = pd.DataFrame([[key, ",".join(value["注文"])]
                       for key, value in order_df.groupby("伝票")],
                      columns=['伝票', '注文']) 
print(result) 
""" =>
  伝票         注文
0  1  ビール,枝豆,刺身
1  2  ウイスキー,チョコ
"""

Comments

[frederick_1974]

ありがとうございます。
このようなやり方を考えたことがなかったので勉強になります。