python - 按函数分组的字符串模式聚合

Question

我有如下所示的数据框

Country  City
UK       London
USA      Washington
UK       London
UK       Manchester
USA      Washington
USA      Chicago

我想对国家进行分组并聚合到一个国家中重复次数最多的城市

我想要的输出应该是这样的

Country City
UK      London
USA     Washington

因为伦敦和华盛顿出现了 2 次，而曼彻斯特和芝加哥只出现了 1 次。

我试过了

from scipy.stats import mode
df_summary = df.groupby('Country')['City'].\
                        apply(lambda x: mode(x)[0][0]).reset_index()

但它似乎不适用于字符串

Answer 1

我无法复制你的错误，但你可以使用 pd.Series.mode , 它接受字符串并返回一个系列，使用 iat提取第一个值:

res = df.groupby('Country')['City'].apply(lambda x: x.mode().iat[0]).reset_index()

print(res)

  Country        City
0      UK      London
1     USA  Washington