替换 csv 标题而不删除其他行

Question

我想替换一个 cvs 文件的标题行text.csv。

header_list = ['column_1', 'column_2', 'column_3']

标题将如下所示；

column_1, column_2, column_3

这是我的代码；

import csv
with open('text.csv', 'w') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(header_list)

csv 文件的标题已正确替换。但是，csv 文件中的其余行已被删除。如何仅替换标题而使其他行保持不变？

我正在使用 python v3.6

Answer 1

这是使用csv 模块的正确方法。
csv.DictReader 将 csv 文件的内容读入字典列表。它采用可选的fieldnames 参数，如果设置该参数，则应用自定义标题并忽略原始标题并将其视为数据行。所以，你需要做的就是阅读你的 csv 使用csv.DictReader 写入文件并使用csv.DictWriter 写入数据。您必须删除reader 中的第一行，因为它包含旧标题并写入新标题。不过，将新数据写入单独的文件确实有意义。

import csv

header = ["column_1", "column_2", "column_3"]

with open('text.csv', 'r') as fp:
    reader = csv.DictReader(fp, fieldnames=header)

    # use newline='' to avoid adding new CR at end of line
    with open('output.csv', 'w', newline='') as fh: 
        writer = csv.DictWriter(fh, fieldnames=reader.fieldnames)
        writer.writeheader()
        header_mapping = next(reader)
        writer.writerows(reader)

Answer 2

使用这个：

import csv
header_list = ['column_1', 'column_2', 'column_3']
mystring = ",".join(header_list)
def line_prepender(filename, line):
    with open(filename, 'r+') as csvfile:
        content = csvfile.read()
        csvfile.seek(0, 0)
        csvfile.write(line.rstrip('\r\n') + '\n' + content)

line_prepender("text.csv", mystring)