python - 将标题行写入 csv python

Question

如何将标题添加到 csv 的第一行？

我的解决方案目前会附加所有内容

类似于:writer.writerow(['DataA', 'DataB', 'DataC', 'DATA D'])[0]

我觉得有一种简单的方法可以做到这一点，但我忽略了显而易见的事情。

我在网上查看了很多示例，但仍然在努力了解如何在 csv 中轻松地做到这一点。

举个例子，假设我想抓取 SO 的数据 - 加载 selenium，将其写入四列并执行 20 页。我想每次将标题写入第 1 行，然后附加抓取的数据

with open('C:\\fa.csv', 'a+', newline='', encoding="utf-8") as outfile:
    writer = csv.writer(outfile)
    writer.writerow(['DataA', 'DataB', 'DataC', 'DataD'])
    for row in zip(ZAW_text, RESULTS1, RESULTS):
        writer.writerow(row)
        #writer.writerows({'Date': row[0], 'temperature 1': row[1], 'temperature 2': 0.0} for row in writer)
        print(row)

Answer 1

import csv
import os

file_name = 'C:\\fa.csv'
fake_data = [[1, 2, 3, 4, 5, 4444, 6, 67], [1, 2, 3, 4, 5, 4444, 6, 67],
             [1, 2, 3, 4, 5, 4444, 6, 67], [1, 2, 3, 4, 5, 4444, 6, 67]]
headers = ['DataA', 'DataB', 'DataC', 'DataD']

if os.path.isfile(file_name):
  with open(file_name, 'a', encoding="utf-8") as outfile:
    writer = csv.writer(outfile)
    for datum in fake_data:
      writer.writerow(datum)
else:
  with open(file_name, 'w', encoding="utf-8") as outfile:
    writer = csv.writer(outfile)
    writer.writerow(headers)
    for datum in fake_data:
      writer.writerow(datum)

在第一行 open('C:\\fa.csv', 'a+', newline='',encoding="utf-8") 你有 a+ 是 append 你想要的 w 是 write。

看看这个:https://docs.python.org/3/tutorial/inputoutput.html#reading-and-writing-files