python - 根据列名称进行计算并获取名称作为计算的一部分

Question

我需要根据列名计算差异，并使用部分列名作为计算的一部分。

目前，促销值的列为空。我需要计算第一行 get2for30 为 (X-30) 和第二列 GET2FOR31 (x-31) 并在相应的“单元格”中显示值。我想知道是否有办法使用 pandas 来做到这一点。

更新 - 我遇到的另一个问题是 Get2forX 的列最多为 90，因此最后一列是 get2for90。我想知道是否有类似函数的东西可以应用于所有这些类型的列。 (无需编写 90 行代码)

Answer 1

你可以简单地这样做:-

df['GET2FOR30'] = df.Prices - 30
df['GET2FOR31'] = df.Prices - 31
... 

或者对于大量列:-

cols = df.columns
initial = 30
for i, j in enumerate(cols[2:]):
    if 'GET2FOR'in j:
        df[j] = df.Prices - (i + initial)