從數據框中獲取字典的有效方法

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()

Answer 2

df.groupby('City')是一個迭代器。 您可以在dict理解中使用它來生成所需的dict：

In [84]: {city:set(grp['Name'].tolist()) for city, grp in df.groupby('City')}
Out[84]: {'Portland': {'Bob', 'Mallory'}, 'Seattle': {'Alice', 'Bob'}}

---

要在“名稱”列中創建帶有集合的DataFrame，可以使用groupby/apply ：

In [91]: df.groupby('City')['Name'].apply(set).reset_index()
Out[91]: 
       City            Name
0  Portland  {Bob, Mallory}
1   Seattle    {Alice, Bob}

Answer 3

你可以做

dict(df.groupby('City')['Name'].apply(list))