gpt4 book ai didi

python - 有条件地填充数据框列的所有后续值

转载 作者:太空宇宙 更新时间:2023-11-03 14:41:44 25 4
gpt4 key购买 nike

我想根据满足条件的第一个实例“前向填充”DataFrame 中新列的值。这是一个基本示例:

import pandas as pd
import numpy as np

x1 = [1,2,4,-3,4,1]

df1 = pd.DataFrame({'x':x1})

我想向 df1 添加一个新列 - 'condition' - 当出现负数时值为 1,否则为 0,但是我想要剩余的一旦找到负数,列的值将设置为 1

所以,我会按如下方式寻找所需的输出:

    condition   x
0 0 1
1 0 2
2 0 4
3 1 -3
4 1 4
5 1 1

最佳答案

到目前为止还没有人使用过cummax:

In [165]: df1["condition"] = (df1["x"] < 0).cummax().astype(int)

In [166]: df1
Out[166]:
x condition
0 1 0
1 2 0
2 4 0
3 -3 1
4 4 1
5 1 1

关于python - 有条件地填充数据框列的所有后续值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52684978/

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