根据条件覆盖列行

Question

现有数据框：

Id           condition1        condition2       score
A               attempt           pass            0
A               attempt           fail            0
B               attempt           pass            0
B               attempt           level_1         0
B               attempt           fail            0
C               attempt           fail            0
D               attempt           fail            0

预期数据框：

Id           condition1        condition2       score
A               attempt           pass            1
A               attempt           fail            1
B               attempt           pass            1
B               attempt           level_1         1
B               attempt           fail            1
C               attempt           fail            0
D               attempt           fail            0

我正在寻找标记分数每一行如果满足以下条件的任何行，则唯一 ID 为 1：condition1 == 'attempt' & condition2 =='pass'。

Answer 1

你可以试试：

m1 = df['condition1'].eq('attempt')
m2 = df['condition2'].eq('pass')

df['score'] = (m1 & m2)
df['score'] = df.groupby('Id')['score'].transform(lambda x: x.any().astype(int))

  Id condition1 condition2  score
0  A    attempt       pass      1
1  A    attempt       fail      1
2  B    attempt       pass      1
3  B    attempt    level_1      1
4  B    attempt       fail      1
5  C    attempt       fail      0
6  D    attempt       fail      0