如何向CSV文件中添加新列？

Question

我有几个CSV文件如下所示：

Input
Name        Code
blackberry  1
wineberry   2
rasberry    1
blueberry   1
mulberry    2

我想添加一个新的列到所有的CSV文件，使它看起来像这样：

Output
Name        Code    Berry
blackberry  1   blackberry
wineberry   2   wineberry
rasberry    1   rasberry
blueberry   1   blueberry
mulberry    2   mulberry

到目前为止，我的剧本是：

import csv
with open(input.csv,'r') as csvinput:
    with open(output.csv, 'w') as csvoutput:
        writer = csv.writer(csvoutput)
        for row in csv.reader(csvinput):
            writer.writerow(row+['Berry'])

（Python 3.2）

但在输出中，脚本跳过每一行，新列中只有Berry：

Output
Name        Code    Berry
blackberry  1   Berry

wineberry   2   Berry

rasberry    1   Berry

blueberry   1   Berry

mulberry    2   Berry

Answer 1

import csv
with open('input.csv','r') as csvinput:
    with open('output.csv', 'w') as csvoutput:
        writer = csv.writer(csvoutput)

        for row in csv.reader(csvinput):
            if row[0] == "Name":
                writer.writerow(row+["Berry"])
            else:
                writer.writerow(row+[row[0]])

也许这样的事情是你想要的？

此外，csv代表逗号分隔的值。所以，你需要逗号来分隔你的价值观，我认为：

Name,Code
blackberry,1
wineberry,2
rasberry,1
blueberry,1
mulberry,2

Answer 2

我很惊讶没有人推荐熊猫。尽管使用一组依赖项（如Pandas）可能看起来比完成如此简单的任务所需的更为繁重，但它生成了一个非常短的脚本，Pandas是执行各种CSV（实际上是所有数据类型）数据操作的伟大库。不能与4行代码争论：

import pandas as pd
csv_input = pd.read_csv('input.csv')
csv_input['Berries'] = csv_input['Name']
csv_input.to_csv('output.csv', index=False)

更多信息，请访问熊猫网站！

output.csv的内容：

Name,Code,Berries
blackberry,1,blackberry
wineberry,2,wineberry
rasberry,1,rasberry
blueberry,1,blueberry
mulberry,2,mulberry

Answer 3

这应该会让你知道该怎么做：

>>> v = open('C:/test/test.csv')
>>> r = csv.reader(v)
>>> row0 = r.next()
>>> row0.append('berry')
>>> print row0
['Name', 'Code', 'berry']
>>> for item in r:
...     item.append(item[0])
...     print item
...     
['blackberry', '1', 'blackberry']
['wineberry', '2', 'wineberry']
['rasberry', '1', 'rasberry']
['blueberry', '1', 'blueberry']
['mulberry', '2', 'mulberry']
>>> 

编辑，注意在py3k你必须使用下一个（r）

谢谢你接受答案。这是您的奖金（您的工作脚本）：

import csv

with open('C:/test/test.csv','r') as csvinput:
    with open('C:/test/output.csv', 'w') as csvoutput:
        writer = csv.writer(csvoutput, lineterminator='\n')
        reader = csv.reader(csvinput)

        all = []
        row = next(reader)
        row.append('Berry')
        all.append(row)

        for row in reader:
            row.append(row[0])
            all.append(row)

        writer.writerows(all)

请注意

>

csv中的lineterminator参数。编写器。默认情况下，它设置为'\r\n'，这就是为什么要使用双间距

使用列表附加所有行并使用writerows一次性写入它们。如果您的文件非常大，这可能不是一个好主意（RAM），但对于普通文件，我认为速度更快，因为I/O更少。

如本文评论所示，请注意，您可以在同一行中使用语句，而不是将这两个嵌套在语句中：

用open（'C：/test/test.csv'，'r'）作为csvinput，open（'C：/test/output.csv'，'w'）作为csv输出：