Добавление нового столбца в pandas df в зависимости от условия - PullRequest
Новая
       18

Добавление нового столбца в pandas df в зависимости от условия

0 голосов
/ 25 февраля 2020

У меня есть следующий набор данных: 

ID   Asset   Boolean
1     "A"    True  
1     "B"    False  
1     "B"    False   
2     "A"    True
3     "A"    True
3     "A"    True
3     "B"    False
3     "B"    False
4     "A"    True
4     "A"    True
5     "A"    True
5     "B"    False

Я хочу добавить еще один столбец, который должен иметь значение True только в том случае, если все значения в столбце Boolean имеют значение True для одного и того же ID. Так что-то вроде этого: 

ID   Asset   Boolean  Check
1     "A"    True     False
1     "B"    False    False
1     "B"    False    False
2     "A"    True     True
3     "A"    True     False
3     "A"    True     False
3     "B"    False    False
3     "B"    False    False
4     "A"    True     True
4     "A"    True     True
5     "A"    True     False
5     "B"    False    False

Я хочу сохранить исходный набор данных для параметров фильтра. Я не мог понять, как перебрать столбец Boolean с учетом столбца ID.

1 Ответ

4 голосов
/ 25 февраля 2020

Вы можете GroupBy и transform с all: 

df['Check'] = df.groupby('ID').Boolean.transform('all')

print(df)

    ID Asset  Boolean  Check
0    1     A     True  False
1    1     B    False  False
2    1     B    False  False
3    2     A     True   True
4    3     A     True  False
5    3     A     True  False
6    3     B    False  False
7    3     B    False  False
8    4     A     True   True
9    4     A     True   True
10   5     A     True  False
11   5     B    False  False
...