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() 

Я не знаю, почему это закончится бесконечным циклом.

EDIT: Я сделал ошибку, и моя оригинальная программа бесконечный цикл имел следующую строку:

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

Я изменил 'a' к 'w', но это в конечном итоге стереть весь CSV сам файл. Так что теперь у меня другая проблема.

Answer 1

У вас есть проблема здесь if row[2] or row[3] or row[4] == '0': и здесь row[9] == 'Label', вы можете использовать any для проверки нескольких переменных, равные тому же значению, и использовать = присвоить значение, также я бы рекомендовал использовать with open.

Кроме того, вы не можете читать и писать в то же время в csv файле, так что вам нужно, чтобы сохранить изменения в новый файл CSV, вы можете удалить оригинал после и переименовать новый, используя os.remove и 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

Я думаю, что проблема в линии

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

Вы открываете тот же файл для чтения и writing.So попробуйте использовать другое имя файла.

Answer 3

вы можете написать в tempfile.NamedTemporaryFile и просто использовать in для проверки "0", как вы сопрягать полную строку не подстроку, так что вы ничего не сэкономить, используя любой, как создать кортеж из трех элементов поэтому вы можете также нарезать или просто протестировать членство вне зависимости от того, вы просто замените исходный файл на 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] также вы имя файла и строку, так что все, что вам нужно пройти.