Pandasで複数の列の要素を引数に取った関数を適用させる

Question

現在以下のようなデータフレームを持っています。

日付(DatatimeIndex)	商品名	売上数	発注数	月売上平均	月売上標準偏差
2017-01-01	...	12	0	10.0	5.2
2017-01-02	...	22	0	10.0	5.2
2017-01-03	...	15	0	10.0	5.2
2017-01-04	...	18	50	10.0	5.2
2017-01-05	...	10	0	10.0	5.2
...	...	...	...	...	...
2017-02-01	...	21	41	11.0	3.2
2017-02-02	...	16	0	11.0	3.2

このデータフレームにイメージとして以下のような関数で計算した値を列を加えたいです。

各行に対し、売上数・月売上平均・月売上標準偏差を引数にとる関数 function(['売上数', '月売上平均', '月売上標準偏差']) で計算した値

日付(DatatimeIndex)	商品名	売上数	発注数	月売上平均	月売上標準偏差	関数で計算した列
2017-01-01	...	12	0	10.0	5.2	...
2017-01-02	...	22	0	10.0	5.2	...
2017-01-03	...	15	0	10.0	5.2	...
2017-01-04	...	18	50	10.0	5.2	...
2017-01-05	...	10	0	10.0	5.2	...
...	...	...	...	...	...	...
2017-02-01	...	21	41	11.0	3.2	...
2017-02-02	...	16	0	11.0	3.2	...

Pandasのapply()が調べたところSeriesのような単一の列にしか適用できないとのことだったので、皆様から知見をいただければと思います。

宜しくお願いします。

Answer 1

行を受け取る関数funcを用意し.apply(func,axis=1)にて可能です。
過去質問python データフレーム　複数の列を条件として関数を適用させたいも参考になるかと思います。

def func(row):
    return row['A'] + row['B'] + row['C']

import pandas as pd
df = pd.DataFrame({'A':[1,2,3],'B':[10,20,30],'C':[100,200,300]})
df['D'] = df.apply( func, axis=1)
print (df)
"""
   A   B    C    D
0  1  10  100  111
1  2  20  200  222
2  3  30  300  333
"""

Comments

[pinocookie]

回答ありがとうございます！

なるほど！列の要素を引数にとるのではなく、データフレームそのものを引数に取ればいんですね！
ありがとうございます