如何动态添加数据帧中某些列的值？

Question

图像中的数据帧

year= 2020 (MAX COLUMN)
lastFifthYear = year - 4
input = '2001509-00'

我想将年份（2020年）和上一个第五年（2016年）之间的所有值相加，其中输入PARTNO=输入

因此，对于输入值，我应该得到4 6 2 3 2（2016 2017 2019 2020），即17

请给我一些密码

Answer 1

示例设置

import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randint(0, 10, (10, 10)),
                  columns=list('ab')+list(range(2,10)))

已解决

#sum where a == 9 columns between 3,6 by rows

df['number'] = df.loc[df['a'].eq(9), 
                      pd.to_numeric(df.columns, errors='coerce')
                        .to_series()
                        .between(3, 6)
                        .values].sum(axis=1)
print(df)


   a  b  2  3  4  5  6  7  8  9  number
0  1  9  9  2  6  0  6  1  4  2     NaN
1  2  3  4  8  7  2  4  0  0  6     NaN
2  2  2  7  4  9  6  7  1  0  0     NaN
3  0  3  5  3  0  4  2  7  2  6     NaN
4  7  7  1  4  7  7  9  7  4  2     NaN
5  9  9  9  0  3  3  3  8  7  7     9.0
6  9  0  5  5  7  9  6  6  5  7    27.0
7  2  1  9  1  9  3  3  4  4  9     NaN
8  4  0  5  9  6  7  3  9  1  6     NaN
9  5  5  0  8  6  4  5  4  7  4     NaN

Answer 2

这里是一些应该工作的代码，但你肯定需要改进你在这里提问的方式：-）

考虑到df是您粘贴为上图的表。

>>> year = 2016
>>> df_new=df.query('INPUT_PARTNO == "2001509-00"').melt(['ACTUAL_GI_YEAR', 'INPUT_PARTNO'],  var_name='year',  value_name='number')

>>> df_new.year=df_new.year.astype(int)

>>> df_new[df_new.year >= year].groupby(['ACTUAL_GI_YEAR','INPUT_PARTNO']).agg({'number' : sum})
                             number
ACTUAL_GI_YEAR INPUT_PARTNO
0              2001509-00        17