CSV写入文件困难

Question

我应该根据条件为我的CSV文件添加特定标签。 CSV文件有10列，第三，第四和第五列是最能影响条件的列，我在第十列添加了我的标签。 我这里的代码以无限循环结束：

import csv
import sys

w = open(sys.argv[1], 'w')
r = open(sys.argv[1], 'r')

reader = csv.reader(r)
writer = csv.writer(w)

for row in reader:
    if row[2] or row[3] or row[4] == '0':
        row[9] == 'Label'
        writer.writerow(row)

w.close()
r.close()

我不知道它为什么会以无限循环结束。

编辑：我犯了一个错误，我原来的无限循环程序有这条线：

w = open(sys.argv[1], 'a')

我将'a'更改为'w'但最终删除了整个CSV文件本身。 所以现在我有一个不同的问题。

Answer 1

if row[2] or row[3] or row[4] == '0':这里有问题if row[2] or row[3] or row[4] == '0':这里有row[9] == 'Label' ，你可以用any来检查几个等于相同的变量值，并使用=来赋值，我也建议使用with open 。

此外，您无法在csv文件中同时读取和写入，因此您需要将更改保存到新的csv文件，您可以删除原始文件并使用os.remove和os.rename重命名新os.rename ：

import csv
import sys    
import os

with open('some_new_file.csv', 'w') as w, open(sys.argv[1], 'r') as r:
    reader, writer = csv.reader(r), csv.writer(w)
    for row in reader:
        if any(x == '0' for x in (row[2], row[3], row[4])):
            row[9] = 'Label'
        writer.writerow(row)

os.remove('{}'.format(sys.argv[1]))
os.rename('some_new_file.csv', '{}'.format(sys.argv[1]))

Answer 2

您可以写入tempfile.NamedTemporaryFile并使用in来测试"0"因为您匹配的是完整字符串而不是子字符串，因此在创建三元素元组时不会使用any来保存任何内容，这样您就可以以及切片或只测试成员资格，然后你只需用shutil.move替换原始文件：

import csv
import sys    
from shutil import move
from tempfile import NamedTemporaryFile

with NamedTemporaryFile("w", dir=".", delete=False) as w, open(sys.argv[1]) as r:
    reader, writer = csv.reader(r), csv.writer(w)
    writer.writerows(row[:-1] + ['Label'] if "0" in row[2:5] else row 
                     for row in reader)

move(w.name, sys.argv[1])

sys.argv[1]也是你的文件名和字符串，这就是你需要传递的全部内容。

Answer 3

我认为这个问题符合要求

w = open(sys.argv[1], 'w')
r = open(sys.argv[1], 'r')

您正在打开相同的文件进行读写。请尝试使用不同的文件名。