来自dicts列表的Pandas DataFrame

时间:2017-02-17 17:52:57

标签: python pandas

我有一个数据结构,它是一个dicts列表列表:

[
    [{'Height': 86, 'Left': 1385, 'Top': 215, 'Width': 86},
     {'Height': 87, 'Left': 865, 'Top': 266, 'Width': 87},
     {'Height': 103, 'Left': 271, 'Top': 506, 'Width': 103}],
    ...
]

我可以将其转换为数据框:

detections[0:1]
df = pd.DataFrame(detections)
pd.DataFrame(df.apply(pd.Series).stack())

哪个收益率:

stacked data frame

这几乎就是我想要的,但是

如何将每个单元格中的字典转换为包含列'左','顶部','宽度' '身高'?

2 个答案:

答案 0 :(得分:7)

要添加到Psidom's answer,列表也可以使用itertools.chain.from_iterable展平。

from itertools import chain

pd.DataFrame(list(chain.from_iterable(detections)))

在我的实验中,对于大量的“块”,这大约是两倍。

In [1]: %timeit [r for d in detections for r in d]
10000 loops, best of 3: 69.9 µs per loop

In [2]: %timeit list(chain.from_iterable(detections))
10000 loops, best of 3: 34 µs per loop

如果实际上希望最终数据框中的索引反映原始分组,则可以使用

完成此操作
pd.DataFrame(detections).stack().apply(pd.Series)
       Height  Left  Top  Width
0   0      86  1385  215     86
    1      87   865  266     87
    2     103   271  506    103
1   0      86  1385  215     86
    1      87   865  266     87
    2     103   271  506    103

您已接近,但在堆叠索引后需要应用pd.Series

答案 1 :(得分:1)

您可以遍历列表,构建数据框列表,然后将它们连接起来:

pd.concat([pd.DataFrame(d) for d in detections])

# Height    Left    Top  Width
#0    86    1385    215     86
#1    87     865    266     87
#2   103     271    506    103

或者,首先展开列表然后调用pd.DataFrame()

pd.DataFrame([r for d in detections for r in d])