在python中向下拖动公式

Question

有没有办法为每个水果计算 True 后 5 年的（乘数 x 值）？为了清楚起见，此处显示了公式

df   

fruit   year     value     Criteria        multiplier        multiplier x value
apple   1950      2
apple   1951      3
apple   1952      3
apple   1953      4
apple   1954      5
apple   1955      4       True           1.4                        (=1.4*4)
apple   1956      7                                                 (=1.4*7)
..
apple   2000      8                                                 (=1.4*8)

banana  1950     333                                                
banana  1951     335 

Answer 1

您可以使用df[df['Criteria'] == True].index 查找所有True 行的索引。 现在您可以使用相应的乘数填充后续行，如下所示：

for i in df[df['Criteria'] == True].index:
  df.loc[i:i+4, 'multiplier'] = df.loc[i]['multiplier']

最后将结果写入multiplier x value 列，如下所示：

df['multiplier x value'] = df['multiplier'] * df['value']

这假设您所有的 emtpy 乘数单元格都填充有 None 或 0。

如果每个水果都需要这个，请使用 groupby 拆分 DataFrame 并在每个子帧上运行循环：

result = pd.DataFrame()
for subframe in [x for _, x in df.groupby('fruit')]:
  for i in subframe[subframe['Criteria'] == True].index:
    subframe.loc[i:i+4, 'multiplier'] = subframe.loc[i]['multiplier']
  result = pd.concat([result, subframe])

result['multiplier x value'] = result['multiplier'] * result['value']