Question

我有一个包含一列词典列表的pandas数据框。

    list_dicts
id  
a1  [{name:'cat'}, {name:'dog'}]
a2  [{name:'toy'}, {name:'boy'}] 
a3  [{name:'jack'},{name:'jill'},{name:'sam'}] 
a4  [{name:'pig'}]

dicts列表中的每个键都是＆＃39; name＆＃39;。我想创建一个与＆＃39; name＆＃39;相关的所有值的列表。键并将新列附加到现有数据框，如下所示。

    list_from_dict
id  
a1  ['cat','dog']
a2  ['toy','boy'] 
a3  ['jack','jill','sam'] 
a4  ['pig']

我怎样才能做到这一点？我理解它可能会使用lambda函数，但不确定如何。

Answer 1

你可以使用列表理解并且只在一行中没有lambda函数来执行此操作：

df['list_from_dict'] = [[x['name'] for x in list_dict] for list_dict in df['list_dicts']]

Answer 2

一种方法是将pd.Series.apply与自定义lambda功能一起使用：

df = pd.DataFrame({'list_dicts': [[{'name': 'cat'}, {'name': 'dog'}],
                                  [{'name': 'toy'}, {'name': 'boy'}],
                                  [{'name': 'jack'}, {'name': 'jill'}, {'name': 'sam'}],
                                  [{'name': 'pig'}]]},
                  index=['a1', 'a2', 'a3', 'a4'])

df['list_dicts'] = df['list_dicts'].apply(lambda x: [next(iter(d.values())) for d in x])

print(df)

           list_dicts
a1         [cat, dog]
a2         [toy, boy]
a3  [jack, jill, sam]
a4              [pig]

将拨号列表解压缩到pandas数据帧中的列表中

2 个答案: