python - 乘以 2 列，直到获得所需的值

Question

向大家问好，

我有这个表(没有 Res_Problem):

<表类=“s-表”> <标题> ID 问题 X 影响 问题 Res_Problem <正文> ID1 12 IDC1 1 2 (12-2)=10 ID1 12 IDC2 2 2 (10-4)=6 停止 ID1 12 IDC3 1 0 无循环 ID1 12 IDC4 1 0 无循环 ID2 10 IDB1 1 2 新循环 (10-2)=8 ID2 10 IDB1 1 2 (8-2) = 6 停止

我想做一个循环，将影响和概率相乘，直到得到所需值(例如 6)，然后停止循环，直到达到 6。但在 ID2 上再次开始循环...等等，有什么建议吗？

我认为它必须是这样的:

while (df['问题'] - df['影响']*df['影响'] < 6): df['循环'] = res

循环应创建“Res_Problem”列

Answer 1

这里有一个选项:

s = (df['Problem']
 .sub(df['Impact'].mul(df['Prob'])
      .groupby(df['ID']).cumsum()
      )
)

m = s.le(6).groupby(df['ID']).shift(fill_value=False)

df['Res_Problem'] = s.mask(m)

输出:

    ID  Problem     X  Impact  Prob  Res_Problem
0  ID1       12  IDC1       1     2         10.0
1  ID1       12  IDC2       2     2          6.0
2  ID1       12  IDC3       1     0          NaN
3  ID1       12  IDC4       1     0          NaN
4  ID2       10  IDB1       1     2          8.0
5  ID2       10  IDB1       1     2          6.0