python - 按 ID 分组并对一列的值应用条件

Question

我是 Pandas 图书馆的新手，非常感谢您的帮助。
这是我的输入


ID
日期
标签
股票


机管局
01/26/2020
真
100

机管局
2020/01/27
错误的
200

BB
2020/01/28
错误的
300

BB
01/29/2020
真
500

BB
01/30/2020
错误的
100

抄送
01/26/2020
真
200

抄送
2020/01/27
错误的
300

抄送
2020/01/28
错误的
100

抄送
01/29/2020
真
400


我想查看 Label 列，对于为每个 ID 找到的第一个“TRUE”标签，我需要在新列中复制股票的值。如果一个 ID 有多个 TRUE，我只需要考虑第一个。
这是我想要获得的输出:


ID
日期
标签
股票
新的


机管局
01/26/2020
真
100
100

机管局
2020/01/27
错误的
200

BB
2020/01/28
错误的
300

BB
01/29/2020
真
500
500

BB
01/30/2020
错误的
100

抄送
01/26/2020
真
200
200

抄送
2020/01/27
错误的
300

抄送
2020/01/28
错误的
100

抄送
01/29/2020
真
400



感谢您的帮助!

Answer 1

另一个版本，使用 .groupby() :

m = df.index.isin(
    df[df["LABEL"] == True].reset_index().groupby("ID").index.first()
)
df.loc[m, "NEW"] = df.loc[m, "STOCK"]
print(df)

打印:

   ID        DATE  LABEL  STOCK    NEW
0  AA  01/26/2020   True    100  100.0
1  AA  01/27/2020  False    200    NaN
2  BB  01/28/2020  False    300    NaN
3  BB  01/29/2020   True    500  500.0
4  BB  01/30/2020  False    100    NaN
5  CC  01/26/2020   True    200  200.0
6  CC  01/27/2020  False    300    NaN
7  CC  01/28/2020  False    100    NaN
8  CC  01/29/2020   True    400    NaN