我正在通过在同一行中添加列来制作一个动态表,然而,列的范围是基于两个列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

enter image description here

我想创建一个SUM_COLUMN,它根据列名中的COLUM_START和COLUM_STOP对列求和.

desired output:

enter image description here

感谢您的帮助!!

推荐答案

对两列之间按列比较的 Select 值使用数字广播,并对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  
    

如果小数据和性能不重要,请使用DataFrame.loc:

c = df.columns.to_numpy()

df['SUM_OF_RANGE'] = [df.loc[i, str(a):str(b)].sum() 
                      for i, a, b in zip(df.index, df['colum_start'], df['colum_stop'])]
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  

Python-3.x相关问答推荐

确定字符串的长度并提取前15或14个字符

Select 作为 MultiIndex 一部分的两个 DatetimeIndex 之间的行

Heroku 中的未知错误代码缺少一个或多个参数

为什么空列表也能起作用?

当我在正则表达式末尾使用斜杠时,为什么会得到不同的结果?

从一列字符串中提取子字符串并将它们放入列表中

平移数组

如何从形状汇总图中提取实际值

通过最接近的匹配合并两个不同长度的列上的两个数据框

FastAPI - 调用 API 时设置 response_model_exclude

每个数据行中每个数据帧值的总和

python 3.7 websockets的1006连接异常关闭错误

Python 异步调试示例

如何用pymongo连接远程mongodb

python 内置的 __exit__ 参数类型是什么?

python 3的蓝牙库

如何使用已打开并使用登录凭据登录的浏览器

try 在 Mac OS 中运行此命令pipenv install requests时出错

带有 Emacs 的 Python 3

为什么异步库比这个 I/O 绑定操作的线程慢?