gpt4 book ai didi

python - Pandas 数据帧 : Calculate Sum based on boolean values in another column

转载 作者:行者123 更新时间:2023-12-01 20:26:52 25 4
gpt4 key购买 nike

我对 Python 相当陌生,我尝试用 pandas 模拟以下逻辑

enter image description here

我当前正在循环抛出行,并希望对前一行中 AMOUNT 列中的值求和,但仅限于最后看到的“TRUE”值。对于实际数据来说似乎效率低下(我有一个大约 500 万行的数据框)?想知道在 Python 中处理此类逻辑的有效方法需要什么?

逻辑:逻辑是,如果 FLAG 为 TRUE,我想对前面行的 AMOUNT 列中的值求和,但仅限于最后看到的“TRUE”值。基本上将 FLAG 为 TRUE 的行之间的“AMOUNT”值相加

最佳答案

检查cumsumtransform sum

df['SUM']=df.groupby(df['FLAG'].cumsum()).Amount.transform('sum').where(df.FLAG)

关于python - Pandas 数据帧 : Calculate Sum based on boolean values in another column,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/62238202/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com