在 Python 中将字典写入 csv 的最有效方法是什么？

Question

我正在编写代码以通过从网站上抓取 html 表来生成 csv 文件。该函数将查看表<tr> 的每一行并将数据列存储在字典中，如下所示

 def write_data():
    table_date = get_data()   # call function to get data from html table into a dict
    // write table_date to csv

 def get_data():
   data = {}
   for row in tr:
      data['name'] = 'John'
      data['id'] = 12
      return data

这是一个简化版本，但本质上我需要一种方法来获取每个表行的字典对象 data 并将其写入 csv，其中的键将是标题行。有什么有效的方法来做到这一点？

Answer 1

使用csv.DictWriter() class；只需将每一行的字典发送给它：

writer = csv.DictWriter(open_writable_file, fieldnames=['id', 'name'])
writer.writeheader()  # write a row the fieldnames

对于您制作的每本词典：

writer.writerow(table_data)

请确保使用newline='' 选项打开可写文件，以让csv 模块控制行尾：

with open(filename, 'w', newline='') as open_writable_file):

出于某种原因，csv.DictWriter 文档示例中省略了此建议；但该对象是csv.writer() class 的子类，那里的建议同样适用。