我有一个数据框如下:
City Name
0 Seattle Alice
1 Seattle Bob
2 Portland Mallory
3 Portland Bob
获取以城市为键、以名称为值的字典的有效方法是什么,即:
dict = {'Seattle': {'Alice','Bob'}, 'Portland': {'Mallory','Bob'}}
我正在考虑使用 df.groupby('City')
请您参考如下方法:
您可以使用 GroupBy.apply 和 to_dict:
df.groupby('City')['Name'].apply(set).to_dict()
