Question

希望任何人都可以帮助我：

我有一个直接写入数据帧的数据库连接。但是，我希望能够将其转换为包含列表的字典列表。因此，在我的示例中，从df到list_of_dict：

import pandas

# FROM:
df = pandas.DataFrame({"ID":["A1","B2","C3","D4"], "x":[1,1,2,2], "Length":[10,11,6,8], "Width":[1,1.3,2,1], "Height": [0.5,0.5,0.5,0.5]})


#TO:
list_of_dict = [{"ID":"A1", "x":1, "Size":[10,1,0.5]},
                {"ID":"B2", "x":1, "Size":[11,1.3,0.5]},
                {"ID":"C3", "x":2, "Size":[6,2,0.5]},
                {"ID":"D4", "x":2, "Size":[8,1,0.5]}]

列表Size包含Length，Width和Height。

如果我们在网上寻找解决方案，但他们似乎并未解决这一具体挑战。由于我是列表和词典的新手，所有的帮助将不胜感激！

Answer 1

这样可以解决问题：

df['Size'] = df[['Length', 'Width', 'Height']].values.tolist()

list_of_dict = df[['ID', 'x', 'Size']].to_dict(orient='records')

Answer 2

df.to_dict(orient = 'records')

orient参数控制输出格式

如何从数据框到包含列表的字典列表

2 个答案: