使用python删除csv文件的单元格内容

Question

我用逗号分隔csv文件，如下所示：

A,B,C,D,E
1,4,5,6,7
9,52,12,0,7
1,5,15,6,9

我想删除第4列的内容，但不删除标题。

所需的输出如下：

A,B,C,D,E
1,4,5,,4
9,52,12,,7
1,5,15,,9

如果不使用熊猫我该怎么办？

Answer 1

首先打开文件，将其附加到csv阅读器对象。

通过在csv.reader对象上手动迭代来读取标题（特例）。然后打开输出文件，读取输入行，删除第4个元素并写回输出文件中的行：

import csv

with open("input.csv") as fr:
    cr = csv.reader(fr)
    title = next(cr)
    with open("output.csv","w") as fw:  # add ,newline='' for python 3
        cw = csv.writer(fw)
        cw.writerow(title)
        for row in cr:
            row[3]=""
            cw.writerow(row)