如何通过python更改CSV文件中的值？

Question

我需要能够找到和更改CSV股票文件中的值。 我知道如何搜索值，但是当我尝试使用csv.writer它将返回错误。

Traceback (most recent call last):
  File "\\pbsrvfile\ExamAccounts$\ITCA76\ochot_krystian_a453\Task 3\Task 3.pyw", line 8, in <module>
    for row in writer:
TypeError: '_csv.writer' object is not iterable

我不知道为什么要这么做。 到目前为止，我的代码是这样的：

import csv
code=input('Enter code to search: ')
stock=open ('Stock Levels.csv', 'wb')
writer=csv.writer(stock)
for row in writer:
    for field in row:
        if field==code:
            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
            writer.writerow[3](restocked_value)
            print(row[1]+' has been restocked to '+restocked_value)

谁能告诉我如何解决该错误，或​​者是否有另一种方法来修改该值。 我正在使用python 3.2.3

Answer 1

尝试这个：

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))
            writer.writerow(row)
os.remove('Stock Levels.csv')
os.rename('New Stock Levels.csv', 'Stock Levels.csv')

如果使用with open而不是open ，则不必关闭文件。

该代码如下所示：

• 读取库存水平（您可能必须将分隔符更改为任何现有的分隔符）
• 查看库存水平的每一行，如果看到要搜索的代码，请对其进行修改
• 将所有库存水平行写回到新的库存水平，包括修改的行
• 删除旧文件以将New Stock Levels.csv文件重命名为旧版本（确保已备份）
• 将New Stock Levels.csv重命名为Stock Levels.csv

我认为您不能直接修改文件，我认为您总是必须先读取，修改，写入临时文件然后重命名。