python csv模块追加列_使用Python在CSV上添加新列

I have the following list of numbers: ['Number', 1,2,3,4]

If I have the following CSV file:

`Name`

`First`

`Second`

`Third`

`Fourth`

How do I add my list of numbers to it and make it look like this:

`Name Number`

`First 1`

`Second 2`

`Third 3`

`Fourth 4`

解决方案

You can use fileinput.input with inplace=True to modify the original file:

import fileinput

import sys

l =['Number', 1,2,3,4]

for ind, line in enumerate(fileinput.input("in.csv",inplace=True)):

sys.stdout.write("{} {}\n".format(line.rstrip(), l[ind]))

Input:

Name

First

Second

Third

Fourth

Output:

Name Number

First 1

Second 2

Third 3

Fourth 4

Or write to a tempfile and move with shutil.move to replace the original file:

l =['Number', 1,2,3,4]

from shutil import move

from tempfile import NamedTemporaryFile

with open('in.csv') as csvfile, NamedTemporaryFile("w",dir=".", delete=False) as temp:

r = csv.reader(csvfile)

wr = csv.writer(temp,delimiter=" ")

for row,new in zip(r,l):

wr.writerow(row+[new])

move(temp.name,"in.csv")