如何将字典列表保存为 CSV 并按顺序保存键？

Question

我有一个字典列表，需要将它们保存为 CSV。第一行是标题列表字段名称。

字段顺序应与标题中的相同。问题是某些字段是可选的，而某些 dicts 不包含它们。另一个问题是 python 不会按照定义的顺序迭代 dict 键。

import csv
import sys

fields = ['first_name', 'last_name', 'city']

dicts = [
    {
        "first_name": "John",
        "last_name": "Doe",
        "city": "New York",
    },
    {
        "first_name": "Agent",
        "last_name": "Smith",
        "city": "Matrix",
    },
]

csv_writer = csv.writer(sys.stdout)

csv_writer.writerow(fields)

for obj in dicts:
    # fields order mismatch
    csv_writer.writerow(obj.values())

这会产生与标题不匹配的 CSV 行。

first_name, last_name, city
New York,   John,      Doe
Matrix,     Agent,     Smith

我可以手动对字段进行排序并检查缺失的项目，但我想知道是否有更好的解决方案可以按预定义的顺序将 dict 值输出到 CSV？

Answer 1

使用 DictWriter（您在 fieldnames 列表中也有错字：fisrt_name 而不是 firs强>t_name):

import csv
import sys

fieldnames = ['first_name', 'last_name', 'city']

dicts = [
    {
        "first_name": "John",
        "last_name": "Doe",
        "city": "New York",
    },
    {
        "first_name": "Agent",
        "last_name": "Smith",
        "city": "Matrix",
    },
]
writer = csv.DictWriter(sys.stdout, fieldnames=fieldnames)

writer.writeheader()
for row in dicts:
    writer.writerow(row)

Answer 2

接受的答案很好，它利用了 python 标准库csv。我只是把我的答案放在这里（使用pandas 方法）以防它可能对某人有所帮助。

In [12]: import pandas as pd

In [13]: dicts = [ 
    ...:     { 
    ...:         "first_name": "John", 
    ...:         "last_name": "Doe", 
    ...:         "city": "New York", 
    ...:     }, 
    ...:     { 
    ...:         "first_name": "Agent", 
    ...:         "last_name": "Smith", 
    ...:         "city": "Matrix", 
    ...:     }, 
    ...: ]                                                                                                                                                                                                         

In [14]: df = pd.DataFrame(dicts)                                                                                                                                                                                  

In [15]: df                                                                                                                                                                                                        
Out[15]: 
  first_name last_name      city
0       John       Doe  New York
1      Agent     Smith    Matrix

In [16]: df.to_csv("/path/to/save", index=False) # if you want to save

希望这会有所帮助。