python - 如何将带有锯齿状数组的字典列表转换为 DataFrame

Question

对 Pandas 相当陌生。我正在尝试获取如下所示的 json 数据集

data = [
    {'id':1, 'colors':['red', 'blue']},
    {'id':2, 'colors':['red', 'blue', 'green']},
    {'id':3, 'colors':['orange', 'blue', 'orange']},
]

进入一个看起来像这样的 Pandas DataFrame

import pandas as pd
df = pd.DataFrame({'id':[1,2,3],
                  'blue':[1,1,1],
                  'green':[0,1,0],
                  'orange':[0,0,2],
                  'red':[1,1,0]})
df
     blue  green  id  orange  red
0     1      0   1       0    1
1     1      1   2       0    1
2     1      0   3       2    0

其中列是“id”和唯一颜色，行是每个原始字典中每种颜色的 id 和计数。我该怎么做？

Answer 1

想通了。

df = [DataFrame(e) for e in data]
df = pd.concat(df)
df = df.pivot_table(index=['id'], columns=['colors'], aggfunc=len).fillna(0)
df

   blue  green  id  orange  red
0     1      0   1       0    1
1     1      1   2       0    1
2     1      0   3       2    0