从具有任意长度值的字典创建数据帧(使用回收键作为列值)

时间:2019-06-01 10:56:18

标签: python dataframe dictionary

我正在努力将字典转换为数据框。

已经有很多答案说明如何以https://stackoverflow.com/a/52819186/6912069之类的“宽格式”进行操作,但是我想做一些不同的事情,最好不要使用循环。

考虑以下示例:

我有一本这样的字典

d_test = {'A': [1, 2], 'B': [3]}

我想得到一个像这样的数据框

index id    values
0     A     1
1     A     2
2     B     3

索引可以是普通的连续整数列。通过回收,我的意思是将'A'=[1, 2]分成两行,其中id列中的A和values列中的值。这样,我将拥有字典项的“长格式”数据框。

这似乎是一件非常基本的事情,但是我想知道是否有一种优雅的pythonic方式来实现这一目标。非常感谢您的帮助。

1 个答案:

答案 0 :(得分:1)

我将创建2个列表。一个来自键,另一个来自字典的值。定义列表后,可以将列表传递到DataFrame中。

import pandas as pd
dic = {'A': [1, 2], 'B': [3], 'D': [4, 5, 6]}

keys = []
values = []

for key, value in dic.items():
    for v in value:
        keys.append(key)
        values.append(v)

df = pd.DataFrame(
    {'id': keys,
     'values': values,
    })
print(df)