Как записать измененный CSV-файл

Question

Перед тем как помечать как дубликат, я прочитал 5 других тем, похожих на это, но, похоже, не работал.

Я смог изменить значение в файле CSV. Однако я не могу понять, как записать весь файл. Я только смог написать обратно измененную строку.

Код:

import csv 

      file=open("stock.csv") 
      stockfile= csv.reader(file) 
      for line in stockfile: 
       if GTIN in line: 
        currentstock= line[2] 
        targetstock = line[3] 
        newstock = (int(currentstock) - int(Quantity)) 
        targetstock = str(targetstock) 
        newstock = str(newstock) 
        line[2]= int(newstock) 
        print(line) 
        with open("output.csv","w") as outfile: 
         newdata = csv.writer(outfile) 
         newdata.writerow(line) 
         if newstock<targetstock: 
          print("stock needs updating...please wait a moment") 
         else: 
          print("stock is fine") 

Stock.csv:

86947367,banana,100,40 
78364721,apple,50,20 
35619833,orange,20,30 
84716491,sweets,200,90 
46389121,chicken,40,10 

output.csv: (только записывает обратно измененную линию, а не остальной части файла)

86947367 banana 1 40 

Я хочу:

86947367,banana,1,40 
    78364721,apple,50,20 
    35619833,orange,20,30 
    84716491,sweets,200,90 
    46389121,chicken,40,10 

спасибо.

Answer 1

Вы перезаписать выходной файл снова и снова здесь:

with open("output.csv","w") as outfile: 

Переместить эту строку в верхней части:

with open("output.csv","w") as outfile, open("stock.csv") as infile: 
    stockfile= csv.reader(infile) 
    newdata = csv.writer(outfile) 
     for line in stockfile: 
      # ... 
      newdata.writerow(line) 
      # .... 

Полный рабочий пример:

import csv 

GTIN = 'banana' 
Quantity = 99 

with open("output.csv","w") as outfile, open("stock.csv") as infile: 
    stockfile= csv.reader(infile) 
    newdata = csv.writer(outfile) 
    for line in stockfile: 
     if GTIN in line: 
      currentstock= line[2] 
      targetstock = line[3] 
      newstock = (int(currentstock) - int(Quantity)) 
      targetstock = str(targetstock) 
      newstock = str(newstock) 
      line[2]= int(newstock) 
      print(line) 
      if newstock<targetstock: 
       print("stock needs updating...please wait a moment") 
      else: 
       print("stock is fine") 
     newdata.writerow(line) 

Выход:

86947367,banana,1,40 
78364721,apple,50,20 
35619833,orange,20,30 
84716491,sweets,200,90 
46389121,chicken,40,10