python - DataFrame 每列乘以参数然后求和

Question

我有一个 Dataframe，其列是 ['a','b','c'] 和一个 Series param 包含三个值，这是 Dataframe 的参数。 param.index 是 ['a','b','c']。我想实现df['a'] * param['a'] + df['b'] * param['b'] + df['c'] * param['c']。因为我的代码中有太多的列和参数。那么有没有什么简洁优雅的代码可以实现呢？

Answer 1

IIUC，这就是你需要的。

Input dataframe df
    a   b   c
0   1   1   1
1   2   2   2
2   3   3   3
3   4   4   4
4   5   5   5

系列 a=[2,20,200]

解决方案

b=df.columns
df1=df[b]*a

输出df1

    a   b   c
0   2   20  200
1   4   40  400
2   6   60  600
3   8   80  800
4   10  100     1000