尝试这个:
import csv
import os
code=input('Enter code to search: ')
with open ('Stock Levels.csv', 'r') as stock:
reader=csv.reader(stock, delimiter=',')
with open ('New Stock Levels.csv', 'w') as newStock:
writer=csv.writer(newStock, delimiter=',')
for row in reader:
if code in row:
print('Record found! \n')
print('Code: '+row[0])
print('Product name: '+row[1])
print('Target level: '+row[2])
target = int(row[2])
print('Current level: '+row[3]+'\n')
current = int(row[3])
restock_amount = target - current
restocked_value = restock_amount + current
row[3] = restocked_value
print(row[1]+' has been restocked to '+str(restocked_value))
wri