仅从数据框中的某些列获取均值和平均值

Question

我有以下数据框。

对于每个时间点（行）A1,A2,A3； A4,5,6; ... 是 3 个重复。我想获取每行 3 个组的平均值和标准差，并将其添加到新的 df 中。

我努力了：

new_df['A1-A3_mean']=np.mean(df[['A1','A2','A3']],axis=1)
new_df['A1-A3_std']=np.std(df[['A1','A2','A3']],axis=1)

哪个有效，但非常手动且耗时。我尝试使用 groupby('Time').agg({'mean','std'}) 但我不知道如何指定它应该始终采用 3 列。理想情况下，结果列将命名为 A1-3_mean / A1-3_stdev

提前致谢！

Answer 1

你可以试试：

N = 3

cols = list(df.drop(columns='time'))

mapper = {c: f'{cols[i//N]}-{cols[i//N+N-1]}' for i,c in enumerate(cols)}

g = df[cols].rename(columns=mapper).groupby(level=0, axis=1)

out = pd.concat({x: g.agg(x) for x in ['mean', 'std']}, axis=1)

输出：

       mean                 std          
      A1-A3     A2-A4     A1-A3     A2-A4
0  4.666667  3.000000  2.886751  2.000000
1  2.666667  4.333333  1.154701  3.214550
2  6.333333  4.333333  2.309401  1.154701