python - 如何从 Pandas 的字典列中提取值？

Question

我有一个数据框，其中每列 {'value': 2343} 都包含字典。我试图摆脱字典并将单元格设置为等于数据框中所有字典的值。

import pandas as pd
df = pd.DataFrame([
    {'node': 'A', 'read': {'value': 2343}, 'write': {'value': 23}},
    {'node': 'B', 'read': {'value': 334}, 'write': {'value': 233444}},
])

我可以将单元格设置为一列的值:

df['read'] = df['read'].apply(lambda x: x['value'])

有没有办法在数据框中的所有列上执行此操作？我应该只遍历所有列吗？

预期的输出应该是一个数据框:

df = pd.DataFrame([
    {'node': 'A', 'read': 2343, 'write': 23},
    {'node': 'B', 'read': 334, 'write': 233444},
])

Answer 1

您可以使用 applymap :

import pandas as pd
df = pd.DataFrame([
    {'node': 'A', 'read': {'value': 2343}, 'write': {'value': 23}},
    {'node': 'B', 'read': {'value': 334}, 'write': {'value': 233444}},
])

cols = ['read', 'write']
df[cols] = df[cols].applymap(lambda x: x['value'])

print(df)

[Out]:

  node  read   write
0    A  2343      23
1    B   334  233444