从 Pandas 循环字典

Question

我正在尝试从 Pandas 表创建 ts 字典，但无法实现。

我的桌子：

         Bool       name       station_id           Coords
0        True       item1          1        19.28152437,47.78960029
1        True       item2          2        20.15190158,46.25528206

我想要的输出：

a= {
    'item1':[ 19.28152437,47.78960029],
    'item2':[20.15190158,46.25528206],
}

我正在使用 iterrows 来做到这一点：

for i, c in allomasok.iterrows():
     //here is the problem, I can't add name of the dict item 

Answer 1

您可以将pd.DataFrame.set_index 与pd.DataFrame.to_dict 结合使用：

df = pd.DataFrame({'name': ['item1', 'item2'],
                   'Coord': ['19.28152437,47.78960029', '20.15190158,46.25528206']})

df['Coord'] = df['Coord'].apply(lambda x: list(map(float, x.split(','))))
d = df.set_index('name')['Coord'].to_dict()

# {'item1': [19.28152437, 47.78960029], 'item2': [20.15190158, 46.25528206]}