Question

我想以特定的方式使用pandas.groupby。给定一个带有两个布尔列的DataFrame（称为col1和col2）和一个id列，我想以下列方式添加一列：

对于每个条目，if（col2为True）和（col1对于具有相同id的任何条目为True）然后分配True。否则为假。

我举了一个简单的例子：

df = pd.DataFrame([[0,1,1,2,2,3,3],[False, False, False, False, False, False, True],[False, True, False, False, True ,True, False]]).transpose()
df.columns = ['id', 'col1', 'col2']

提供以下DataFrame：

     id   col1   col2
0    0   False   False
1    1   False   True
2    1   False   False
3    2   False   False
4    2   False   True
5    3   False   True
6    3   True    False

根据上述规则，应添加以下列：

0    False
1    False
2    False
3    False
4    False
5     True
6    False

有关优雅方式的任何想法吗？

Answer 1

此代码将生成您请求的输出：

df2 = df.merge(df.groupby('id')['col1'] # group on "id" and select 'col1'
                    .any()              # True if any items are True
                    .rename('cond2')    # name Series 'cond2'
                    .to_frame()         # make a dataframe for merging
                    .reset_index())     # reset_index to get id column back
print(df2.col2 & df2.cond2)             # True when 'col2' and 'cond2' are True

Answer 2

df.groupby('id').col1.transform('any') & df.col2

0    False
1    False
2    False
3    False
4    False
5     True
6    False
dtype: bool

在pandas中使用groupby进行布尔运算

2 个答案: