Python Pandas：表示新列中的每 n 行重复 n 次

Question

我想每三行取列的平均值c2，并将结果保存在新列中c3，以便每个平均值重复三次。这段代码可以完成这个工作：import pandas as pddf = pd.DataFrame({'c1': ['A', 'B','C','D','E','F'], 'c2': [1, 2, 3,3,4,5]})nrow=3temp=df['c2'].rolling(nrow).mean()      #Take rolling meantemp= temp[nrow-1::nrow]                #Select mean value every 3 rowstemp=temp.loc[temp.index.repeat(nrow)]  #Repeat each mean value 3 timestemp.index = range(0,len(df))           #Fix index df['c3']=tempprint(df)结果应c3为 [2,2,2,4,4,4] 列。还有比这5行代码更简单的方法吗？

Answer 1

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