Skip to main content
 首页 » 编程设计

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

2025年05月04日1498°冷暖

我有一个数据框如下:

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