根据其他行的值替换数据框行中的变量

Question

我有一个包含 ID、公式和从公式列中提取 ID 的依赖 ID 列的数据框。现在我必须将所有依赖 ID 替换为基于数据框的公式。

我的方法是运行每行的嵌套循环，以使用替换函数替换公式中的相关 ID。循环将停止，直到没有更多可能的替代。但是我不知道从哪里开始，也不确定这是否是正确的方法。

请问有没有什么功能可以让这个过程更简单？

这是创建当前数据框的代码：

data = pd.DataFrame({'ID':['A1','A3','B2','C2','D3','E3'],
    'Formula':['C2/500','If B2 >10 then  (B2*D3) + 100 else D3+10','E3/2 +20','E3/2 +20','var_i','var_x'],
    'Dependent ID':['C2','B2, D3','E3','D3, E3', '','']})

以下是我当前数据框的示例和我想要的最终结果。当前数据框：

期望最终结果：

Answer 1

用公式递归替换公式中的依赖 ID：

df = pd.DataFrame({'ID':['A1','A3','B2','C2','D3','E3'],
    'Formula':['C2/500','If B2 >10 then  (B2*D3) + 100 else D3+10','E3/2 +20','D3+E3','var_i','var_x'],
    'Dependent ID':['C2','B2,D3','E3','D3,E3', '','']})

def find_formula(formula:str, ids:str):
    #replace all the ids inside formula with the correct formula
    if ids == '':
        return formula
    ids = ids.split(',')
    for x in ids:
        sub_formula = df.loc[df['ID']==x, 'Formula'].values[0]
        sub_id = df.loc[df['ID']==x, 'Dependent ID'].values[0]
        formula = formula.replace(x, find_formula(sub_formula, sub_id))
    return formula

df['new_formula']=df.apply(lambda x: find_formula(x['Formula'], x['Dependent ID']), axis=1)

输出：

ID  Formula Dependent ID    new_formula
0   A1  C2/500      C2      var_i+var_x/500
1   A3  If B2 >10 then (B2*D3) + ...    If var_x/2 +20 >10 then (var_x/2 +20*var_i) + ...
2   B2  E3/2 +20    E3      var_x/2 +20
3   C2  D3+E3       D3,E3   var_i+var_x
4   D3  var_i               var_i
5   E3  var_x               var_x

根据其他行的值替换数据框行中的变量

Substituting variable in a dataframe row based on other row's value

python

pandas

variable-substitution