熊猫按两列分组，并从第三列输出值

Question

colour    num    accepted  returned
grey      1      yes       no
red       2      no        no
grey      4      yes       yes

我上面有数据框，想输出colour和num列的唯一组合，以及returned中的相应值，如下所示

colour    num     returned
grey      1       no    
red       2       no      
grey      4       yes     

使用df.groupby(['colour', 'num']).size()给了我独特的组合，但没有returned列。

Answer 1

如果您确定颜色和数字的组合是唯一的，则可以执行以下操作：

df.groupby(['colour', 'num'])['returned'].max()

当然，如果它不是唯一的，并且同时存在“是”和“否”，则将返回“是”，因为“是”>“否” ...

但是，实际上，该解决方案除了df[['colour','num','returned']].drop_duplicates()之外没有其他任何东西，它肯定更精简。

如果您以某种方式知道行可以重复，但是返回的值是唯一的，并且您希望同时显示行的次数和“返回”中的唯一值，则可以一口气：

df.groupby(['colour','num'])['returned'].agg(['size','max'])

哪个会返回：

            size    max
colour  num     
grey    1   1       no
        4   1       yes
red     2   1       no

Answer 2

根据您的描述，我认为您也应该将returned列分组。

df.groupby(['colour','num','returned']).size() 这将显示每种返回状态的发生次数，按num和colour分组：

colour  num  returned
grey    1    no          1
        4    yes         1
red     2    no          1