使用python替换csv文件中的数据

Question

这是新的输入文件格式。我需要使用 python 自动替换 .csv 文件中一列内容的过程。我也可以使用记事本打开 .csv 文件并替换列的内容，但文件非常大而且需要很长时间。

Name                          ID                                                class  Num
"kanika",""University ISD_po.log";" University     /projects/asd/new/high/sde"","MBA","12"
"Ambika",""University ISD_po.log";" University     /projects/asd/new/high/sde"","MS","13"

在上面，我需要替换ID列的内容。 ID列非常不一致，因为它的内容中有很大的空格和（;，/）等符号。ID列中的新内容应该是“输入”。

此 Id 列用 2 个双引号括起来，并且还有一些额外的空格。而其他列只有 1 个双引号。

有没有办法在 python 中做到这一点？

Answer 1

您可以使用 Python 中的 csv 模块来实现这一点。

csv.reader 将每一行作为字符串列表返回。然后，您可以使用 csv.writer 流式传输每一行并在此时修改 ID 列，但这将创建一个新文件。

所以：

import csv
reader = csv.reader(open('file.csv', 'rb'))
writer = csv.writer(open('outfile.csv','wb'))
for row in reader:
    writer.writerow([row[0], "input", row[2], row[3]])

Answer 2

逐行读取 .csv，在 , 上拆分，然后将第二列替换为“输入”。 边写边写（到不同的文件）：

f = open('mycsv.csv','rb')
fo = open('out.csv','wb')

# go through each line of the file
for line in f:
    bits = line.split(',')
    # change second column
    bits[1] = '"input"'
    # join it back together and write it out
    fo.write( ','.join(bits) )

f.close()
fo.close()

然后你可以根据需要重命名替换原来的文件。