python - 在 pandas 数据框单元格中插入列表

Question

我有一本字典，其中每个键都有一个值列表。 与每个键关联的列表的长度是不同的。 我想将字典转换为具有两列“键”和“值”的 pandas 数据框。每一行的“键”列中有一个字典键，“值”列中有与其关联的值列表。数据框将如下所示:

mapping_dict = {'A':['a', 'b', 'c', 'd'], 'B':['aa', 'bb', 'cc']}

df = 
    Key   Value
0   A     ['a', 'b', 'c', 'd']
1   B     ['aa', 'bb', 'cc']

我尝试使用提供的答案 here根据我的用例修改它。 但它没有输出所需的答案。

Answer 1

如果您传递一个列表，pandas 会将其视为多行。但是，您可以通过将列表放置为外部列表的单个元素来欺骗它，如下所示:

import pandas as pd
mapping_dict = {'A':[['a', 'b', 'c', 'd']], 'B':[['aa', 'bb', 'cc']]}
df = pd.DataFrame(mapping_dict)
df

        A                 B
0   [a, b, c, d]    [aa, bb, cc]