【pandas】groupby後に複数の値をreturnする関数をapplyしたときの結果を配列ではなく複数のれるで得たい

Question

前提

Python3
Jupyter notebook
pandas

実現したいこと

次のようなPandasのDataFrameを持っていて

data1	data2	data3	group
100	110	120	1
130	140	150	1
160	170	180	1
190	200	210	2
220	230	240	2
250	260	270	2
280	290	300	2
310	320	330	3
340	350	360	3

これに対し

def hoge_func(group)
    ・・・
    return num1、num2、num3
result = df.groupby('group').apply(hoge_func)

としました。
hoge_funcは見ての通りdata1、data2、data3の3つの値を使って
新しくnum1、num2、num3を返す関数です。

このとき下のようにresultがpandasのSeriesの形式になってしまい、
一つの列にnum1、num2、num3がタプルとして格納されてしまいます。

index	0
1	(100,200,300)
2	(400,500,600)
3	(700,800,900)

num1、num2、num3を別々の列とした一つのDataFrameを得るにはどうすればよいでしょうか。
ついでにそのDataFrameの列名もnum1、num2、num3としたいのですが
どなたか教えていただけないでしょうか。

試したこと

できてしまったSeriesにたいして、

result.apply(pd.Series)

として改めてDataFrameにすることはできたのですが、
この場合どの時点で列名前を指定するのか分かりませんでした。
あとから.renameで列名を変えるのでもいいのですが、
せっかくならもっとスマートに結果の出力の時点で列名付きのDataFrameを得られないのかなと思いました。

Answer 1

applyする関数で、pandas.Series を返すようにすればいいです。
Seriesのindexが、出来上がるデータフレームの列名になります。

def hoge_func(group)
    ...
    return pd.Series([num1, num2, num3],
                     index=['num1', 'num2', 'num3'])

result = df.groupby('group').apply(hoge_func)

Comments

[beluga00nm]

とてもシンプルな解決方法をありがとうございます！
まるほど、returnの時点でseries形式にしてしまえばよかったのですね。

Answer 2

result = [g.filter(regex='^data', axis=1)
           .rename(columns=lambda x: x.replace('data', 'num'))
           .reset_index(drop=True)
          for _, g in df.groupby('group')]

print(len(result))
print(type(result[0]))
print(result[0])
print(result[1])
print(result[2])

#
3
<class 'pandas.core.frame.DataFrame'>

   num1  num2  num3
0   100   110   120
1   130   140   150
2   160   170   180

   num1  num2  num3
0   190   200   210
1   220   230   240
2   250   260   270
3   280   290   300

   num1  num2  num3
0   310   320   330
1   340   350   360