在Python中覆盖/更改CSV上的字段

Question

没有太多的Python（3.4）的经验，但我在一个程序，这将允许您添加牌照和编辑'状态'。这是一个停车场程序，所以状态是In/Out。你知道吗

我唯一的问题是，我不知道如何在输入之后编辑CSV文件上的特定字段。例：从里到外。你知道吗

例如CSV：

NH08OTR, 2008, Vauxhall, Corsa, Blue, In

我想能够将最后一个字段从'In'更改为'Out'，但是CSV的行数是可变的，所以我只希望它在特定的车牌上执行。示例如下：

Please choose a number plate to change the status of: NH08OTR
Status change: Out
Status successfully changed to 'Out'

然后我希望它把CSV上的“In”改为out。你知道吗

希望你能理解，谢谢你的阅读。你知道吗

Answer 1

假设您有一个包含内容的文件plate_status.csv

""" 
NH08OTR, 2008, Vauxhall, Corsa, Blue, In
NH0873R, 2004, Vauxhall, Corsa, Red, Out
...
"""

等等。只要文件不是太大，我会先读入整个文件，然后更新和覆盖：

import csv

def update_status( filename, plate, status ):
    # read and update
    with open( filename) as f:
        reader = csv.reader( f, delimiter=',' )
        line_dict = dict([ ( x[0].strip() , map(lambda y:y.strip() ,x[1:])) for x in reader] )
        line_dict[ plate ][-1] = status

    # save new
    with open( 'plate_status.txt', 'w') as f:
        writer = csv.writer(f )
        new_file_rows = map( lambda x : [ x[0] ]+ x[1], line_dict.iteritems() )
        writer.writerows( new_file_rows )            

update_status( 'plate_status.csv', 'NH08OTR', 'Out' )

现在文件plate_status.csv读取：

""" 
NH08OTR, 2008, Vauxhall, Corsa, Blue, Out
NH0873R, 2004, Vauxhall, Corsa, Red, Out
...
"""