如何向 csv 文件添加新数据列

Question

我正在将 csv 文件读取到变量 data 中，如下所示：

def get_file(start_file): #opens original file, reads it to array
    with open(start_file,'rb') as f:
        data=list(csv.reader(f))

data 是整个 csv 文件。如何向该数据添加额外的列？

Answer 1

简短回答：在您最喜欢的编辑器中打开 CSV 文件并添加一列

长回答：首先，您应该使用 csv.reader 读取文件；它处理分割行、格式错误的行、引号引起来的所有细节。然后，将阅读器包装在另一种添加列的方法中。

def get_file( start_file )
    f = csv.reader( start_file )
        def data( csvfile ):
            for line in csvfile:
                 yield line + [ "your_column" ]
    return data( f )

编辑

正如您在其他问题中所问的那样：

def get_file( start_file )
    f = csv.reader( start_file )
        def data( csvfile ):
            for line in csvfile:
                line[ 1 ] += "butter"
                yield line
    return data( f )

您使用的就像

lines = get_file( "my_file.csv" )
for line in lines:
    # do stuff

Answer 2

data 是字符串列表的列表，因此...：

for row, itm in zip(data, column):
  row.append(str(itm))

当然，您需要 column 的长度正确，因此您可能需要检查这一点，例如提出一个 exc if len(数据)!= len(列)。

Answer 3

.append 一个值到 data 中每行的末尾，然后使用 csv.writer 写入更新的数据。