Python 3 pandasライブラリtransformの使い方
import pandas as pd
import numpy as np
A=np.array([[1,2,3,4,5],[2,1,1,2,2],[1,2,3,4,5],[2,1,1,2,2],[1,2,3,4,5]])
data=pd.DataFrame(A,index=['li','chen','wang','zhao','qian'],columns=['a','b','c','d','e'])
print(data)
a b c d e
li 1 2 3 4 5
chen 2 1 1 2 2
wang 1 2 3 4 5
zhao 2 1 1 2 2
qian 1 2 3 4 5
key=['ss','kk','kk','ss','ss'] # index
print(data.groupby(key).mean()) #mean key
a b c d e
kk 1.500000 1.500000 2.000000 3.000000 3.5
ss 1.333333 1.666667 2.333333 3.333333 4.0
data内の各要素位置の取値はtransform関数のパラメータ関数によって計算される
print(data.groupby(key).transform(np.mean))
#data
a b c d e
li 1.333333 1.666667 2.333333 3.333333 4.0
chen 1.500000 1.500000 2.000000 3.000000 3.5
wang 1.500000 1.500000 2.000000 3.000000 3.5
zhao 1.333333 1.666667 2.333333 3.333333 4.0
qian 1.333333 1.666667 2.333333 3.333333 4.0
生成されたtsfでは、各位置要素の値はdataの「対応位置要素」がtransformの「関数パラメータ」で演算される(ここでは「対応要素」から「対応パケット列の平均値」を減算する).xの値はdataの各位置要素であり、x.meanのmean方程式の作用範囲はkeyによって決定される
my_transform = lambda x : x-x.mean()
tsf=data.groupby(key).transform(my_transform)
print(tsf)
a b c d e
li -0.333333 0.333333 0.666667 0.666667 1.0
chen 0.500000 -0.500000 -1.000000 -1.000000 -1.5
wang -0.500000 0.500000 1.000000 1.000000 1.5
zhao 0.666667 -0.666667 -1.333333 -1.333333 -2.0
qian -0.333333 0.333333 0.666667 0.666667 1.0