pandasでグループごとに同じ値を追加したい

Question

key	flag
a	False
b	False
a	True
b	False
a	False

このようなデータフレームに対して、

key	flag	flag2
a	False	True
b	False	False
a	True	True
b	False	False
a	False	True

key列でグループ分けし、flag列に一つでもTrueが存在すればTrue、そうでなければFalseをflag2列としてデータフレームへ追加したいのですが、さっぱりわかりません。どうすればいいのでしょうか。

Comments

[x98000]

元の行を全て残す必要があるのでしょうか。key毎にflag==Trueが一つ以上あるかをチェックするという考え方ではダメですか？

[meg_]

抽出とForループで地道に処理してはどうでしょうか？

Answer 1

df['flag2'] = df.groupby('key')['flag'].transform(lambda d: d.any()) で良いと思います

import pandas as pd

df = pd.DataFrame({'key':['a','b','a','b','a'],
                   'flag':[False, False, True, False, False]})
df['flag2'] = df.groupby('key')['flag'].transform(lambda d: d.any())
print(df)
#  key   flag  flag2
#0   a  False   True
#1   b  False  False
#2   a   True   True
#3   b  False  False
#4   a  False   True

Comments

[退会済みユーザー]

回答ありがとうございます(*´ω｀*) 完璧に動作しました。