Python Pandas : Tableau croisé dynamique : aggfunc concatenate au lieu de np.size ou np.sum

Question

J'ai quelques entrées dans le cadre de données comme :

name, age, phonenumber
 A,10, Phone1
 A,10,Phone2
 B,21,PhoneB1
 B,21,PhoneB2
 C,23,PhoneC

Voici ce que j'essaie d'obtenir comme résultat du tableau croisé dynamique :

 name, age, phonenumbers, phonenocount
 A,10, "Phone1,Phone2" , 2
 B,21,  "PhoneB1,PhoneB2", 2
 C,23, "PhoneC" , 1

J'essayais quelque chose comme :

pd.pivot_table(phonedf, index=['name','age','phonenumbers'], values=['phonenumbers'], aggfunc=np.size)

Cependant, je veux que les numéros de téléphone soient concaténés dans le cadre de l'aggfunc. Des suggestions ?

Answer 1

Vous pouvez utiliser agg après la fonction groupby :

df.groupby(['name', 'age'])['phonenumber'].\
    agg({'phonecount': pd.Series.nunique, 
         'phonenumber': lambda x: ','.join(x)
        }
       )

#               phonenumber  phonecount
# name  age     
#    A   10   Phone1,Phone2           2
#    B   21 PhoneB1,PhoneB2           2
#    C   23          PhoneC           1

Ou une version plus courte selon @Root et @Jon Clements :

df.groupby(['name', 'age'])['phonenumber'].\
   agg({'phonecount': 'nunique', 'phonenumber': ','.join})