c2
3行ごとに列の平均を取り、c3
各平均が3回繰り返されるように、結果を新しい列に保存したいと思います。このコードは仕事をすることができます:
import pandas as pd
df = pd.DataFrame({'c1': ['A', 'B','C','D','E','F'], 'c2': [1, 2, 3,3,4,5]})
nrow=3
temp=df['c2'].rolling(nrow).mean() #Take rolling mean
temp= temp[nrow-1::nrow] #Select mean value every 3 rows
temp=temp.loc[temp.index.repeat(nrow)] #Repeat each mean value 3 times
temp.index = range(0,len(df)) #Fix index
df['c3']=temp
print(df)
結果はc3
[2,2,2,4,4,4]の列になります。この5行のコードよりも簡単な方法はありますか?
GroupBy.transform
インデックスによる整数除算またはDataFrame
:の長さによるヘルパー配列で使用します。
nrow = 3
#if default RangeIndex
df['c3'] = df.groupby(df.index // nrow)['c2'].transform('mean')
#alternative if not default RangeIndex
#df['c3'] = df.groupby(np.arange(df) // nrow)['c2'].transform('mean')
print(df)
c1 c2 c3
0 A 1 2
1 B 2 2
2 C 3 2
3 D 3 4
4 E 4 4
5 F 5 4
この記事はインターネットから収集されたものであり、転載の際にはソースを示してください。
侵害の場合は、連絡してください[email protected]
コメントを追加