我正在通过在同一行中添加列来制作动态表,但是,列的范围是根据两列 colum_start 和 colum_stop 确定的:
df = pd.DataFrame({
'Name': ['Apple', 'Banana', 'Orange', 'Cherry', 'Egg', 'Cheese'],
'ID': ['F1', 'F1', 'F1', 'F1', 'V1', 'V2'],
'202101': [1, 10, 20, 30, 40, 50],
'202102': [20, 15, 12, 18, 32, 12],
'202103': [3, 11, 25, 32, 13, 4],
'202104': [32, 11, 9, 82, 2, 1],
'202105': [9, 5, 11, 11, 2, 5],
'colum_start ': [202102, 202101, 202102, 202103, 202101, 202103],
'colum_stop': [202105, 202103, 202105, 202104, 202102, 202105],
})
df
我想创建一个 sum_column,根据列名称中的 colum_start 和 colum_stop 对列进行求和。
所需输出:
感谢您的帮助!!
使用 numpy 广播通过按列比较来选择两列之间的值,并比较链式累积和
1
:
a = df['colum_start'].astype(str).to_numpy()[:, None]
b = df['colum_stop'].astype(str).to_numpy()[:, None]
c = df.columns.to_numpy()
mask = (np.cumsum(c == a, axis=1) == 1) & (np.cumsum(c[::-1] == b, axis=1)[:, ::-1] == 1)
df['SUM_OF_RANGE'] = df.where(mask, 0).sum(axis=1)
print (df)
Name ID 202101 202102 202103 202104 202105 colum_start \
0 Apple F1 1 20 3 32 9 202102
1 Banana F1 10 15 11 11 5 202101
2 Orange F1 20 12 25 9 11 202102
3 Cherry F1 30 18 32 82 11 202103
4 Egg V1 40 32 13 2 2 202101
5 Cheese V2 50 12 4 1 5 202103
colum_stop SUM_OF_RANGE
0 202105 64
1 202103 36
2 202105 57
3 202104 114
4 202102 72
5 202105 10