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':這裏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"因爲你是相匹配的完整的字符串不是一個字符串，從而創建三個元素的元組，你將不會被使用任何保存任何所以你可以切片或者只是測試成員資格，然後你只需用替換原始文件：

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]也是你的文件名和一個字符串，所以這是所有你需要傳遞。