python - 从数据框中获取字典的有效方法

Question

我有一个数据框如下:

           City     Name
    0   Seattle    Alice
    1   Seattle      Bob
    2  Portland  Mallory       
    3  Portland      Bob

获取以城市为键、以名称为值的字典的有效方法是什么，即:

dict = {'Seattle': {'Alice','Bob'}, 'Portland': {'Mallory','Bob'}}

我正在考虑使用 df.groupby('City')

Answer 1

您可以使用 GroupBy.apply 和 to_dict:

df.groupby('City')['Name'].apply(set).to_dict()