Pandasで条件に応じてshiftした値の代入

Question

judgeの列がFalseのときはそのままで、Trueのときは上の行を見て一番近いFalseのときの値を入れるというデータ処理をしたいのですが、うまくいきません。
具体的には、

t	judge	target
2019-12-18 01:30:00	False	0
2019-12-18 01:45:00	True	1
2019-12-18 02:00:00	True	2
2019-12-18 02:15:00	False	2
2019-12-18 02:30:00	True	1

というPandasを(tがindexでDatetime、judgeはbool、targetはintです)

t	judge	target
2019-12-18 01:30:00	False	0
2019-12-18 01:45:00	True	0
2019-12-18 02:00:00	True	0
2019-12-18 02:15:00	False	2
2019-12-18 02:30:00	True	2

のようにしたいです。

試したこと

df.loc[df['target']==True, 'target'] = df['target'].shift(1)

としてみたのですが、
locはpandasの上から順に実行しているのではなく、すべての行を同時に実行したような結果になりうまくいきませんでした。

Answer 1

print(df)
#    judge  target
# 0  False       0
# 1   True       1
# 2   True       2
# 3  False       2
# 4   True       1


df['target'] = df['target'].mask(df['judge']).fillna(method='ffill').astype(int)
print(df)
#    judge  target
# 0  False       0
# 1   True       0
# 2   True       0
# 3  False       2
# 4   True       2

Comments

[LightNov]

ありがとうございました。いったんNaNにするのですね、勉強になります。