Python 3: CSV模块

Question

我正在处理一个简单的csv文件，并想知道如何使用脚本生成的数据更新每行特定单元格中包含的值。

column1, column2, colum3, column4,
bob, 20, blue, hammer
jane, 30, red, pencil
chris, 40, green, ruler

然后：

new_colour = [pink, yellow, black]

是否有方法获取列表并将每个列表项写入csv文件中的colum3下的值中？它的结局如下：

column1, column2, colum3, column4,
bob, 20, pink, hammer
jane, 30, yellow, pencil
chris, 40, black, ruler

谢谢

Answer 1

一个(可能是未优化的)解决方案可能是使用熊猫模块，只要您的CSV文件不太大：

PATH_TO_CSV = <your_path>
new_colour = ['pink', 'yellow', 'black']

df = pd.read_csv(PATH_TO_CSV)
df['colum3'] = pd.Series(new_colour)
df.to_csv(PATH_TO_CSV)