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

我有一个数据框，如下所示：

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

关于python - 从数据框中获取字典的有效方法，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/38001612/