主页 / coding / 问题 / 78925432
Accepted
Bijan
Asked: 2024-08-29 06:04:24 +0800 CST

Pandas 将字典列表解压为列

  • 772

我有一个数据框，其中有一个名为的字段，fields它是字典列表（所有行都具有相同的格式）。数据框的结构如下：

formId    fields
   123    [{'number': 1, 'label': 'Last Name', 'value': 'Doe'}, {'number': 2, 'label': 'First Name', 'value': 'John'}]

我正在尝试解开该fields列，因此它看起来像：

formId    Last Name    First Name
   123          Doe          John

我目前拥有的代码是：

for i,r in df.iterrows():
    for field in r['fields']:
        df.at[i, field['label']] = field['value']

但是这似乎不是最有效的方法。有没有更好的方法来实现这一点？

python

3 个回答

  1. 您可以使用 .apply 和 .concat 将字典转换为系列。最后使用 .pivot 将列转换为标题。

    数据：

    import pandas as pd


data = {"formId": 123, "fields": [{'number': 1, 'label': 'Last Name', 'value': 'Doe'},
                                  {'number': 2, 'label': 'First Name', 'value': 'John'}]}
df = pd.DataFrame(data=data)

    代码：

    df = (pd
      .concat(objs=[df, df.pop("fields").apply(func=pd.Series)], axis=1)
      .pivot(index="formId", columns="label", values="value")
      .reset_index()
      .rename_axis(mapper=None, axis=1)
      )

print(df)

    输出：

       formId First Name Last Name
0     123       John       Doe
    • 2
  2. Best Answer

    就我个人而言，我会构建新的数据框：

    df = pd.DataFrame(
    [
        {"formId": form_id, **{f["label"]: f["value"] for f in fields}}
        for form_id, fields in zip(df["formId"], df["fields"])
    ]
)

print(df)

    印刷：

       formId Last Name First Name
0     123       Doe       John
    • 2

  3. 解决方案：

    v = pd.concat([pd.json_normalize(x) for x in df["fields"]]).pivot_table(
    columns="label", values="value", aggfunc=list
)
out = df[["formId"]].join(v.explode(v.columns.tolist(), ignore_index=True))

    使用来自 OP 的示例输入的结果：

       formId First Name Last Name
0     123       John       Doe

    也适用于多行。对于此示例输入：

    df = pd.DataFrame(
    {
        "formId": [123, 456],
        "fields": [
            [
                {"number": 1, "label": "Last Name", "value": "Doe"},
                {"number": 2, "label": "First Name", "value": "John"},
            ],
            [
                {"number": 1, "label": "Last Name", "value": "Smith"},
                {"number": 2, "label": "First Name", "value": "Jack"},
            ],
        ],
    }
)

    结果：

       formId First Name Last Name
0     123       John       Doe
1     456       Jack     Smith
    • 2

相关问题