Dataframe count of unique values in two columns

Question

I have a big dataframe with two columns A and B:

        A              B                                         
0     US1           URL1                                                                                              
1     US2           URL1                                                                                              
2     US1           URL2
3     US3           URL3
4     US2           URL3
4     US2           URL3

I'd like to obtain a dataframe in which for each unique element in B the count of the unique elements in B and the number of unique element in A associated with b:

         B            Imp      Aud                                       
0     URL1              2        2                                                                                            
1     URL2              1        1                                                                                          
2     URL3              3        2

akuiper · Accepted Answer · 2017-06-28 16:37:32Z

1

You can use groupby.agg; with size to get the number of elements and nunique to get the number of unique elements:

df.groupby('B', as_index=False).A.agg({'Imp': 'size', 'Aud': 'nunique'})

#      B    Imp   Aud
#0  URL1      2     2
#1  URL2      1     1
#2  URL3      3     2

answered Jun 28, 2017 at 16:37

akuiper

216k33 gold badges362 silver badges379 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

user3620915 Over a year ago

Thank you @Psidom !

Collectives™ on Stack Overflow

Dataframe count of unique values in two columns

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related