python - 多个 DF 列上的 Sumproduct

Question

ID    height_1    area_1   in_1    height_2    area_2   in_2    height_3    area_3   in_3
1     4           20       1       7           87       1       2          21       1
2     10          83       1       12          32       1       9          41       0
3     16          78       1       12          17       0       np.nan     np.nan   np.nan

如何在通用方法中为每一行计算 sumproduct 看起来像这样......？

sumproduct = height_1 * area_1 * in_1 + height_2 * area_2 * in_2 + height_3 * area_3 * in_3 + ...  ```

例如 row1 = 4 * 20 * 1 + 7 * 87 * 1 + 2 * 21 * 1 我的数据框有大约 20 倍`height_、area_、in_

Answer 1

使用 wide_to_long和 product和 sum

s = (pd.wide_to_long(df, i='ID', j='val', 
                     stubnames=['height','area','in'] ,sep='_', suffix='\d+')
       .product(1).sum(level=0))

Out[575]:
ID
1     731.0
2    1214.0
3    1248.0
dtype: float64

将其分配给 df :

df['sumproduct'] = df.ID.map(s)

或者

df['sumproduct'] = s.values

或者

df['sumproduct'] = s.to_numpy()