复制CSV文件的一列并使用python将其添加到另一个文件中

Question

我有两个文件，第一个文件名为book1.csv，如下所示：

 header1,header2,header3,header4,header5
 1,2,3,4,5
 1,2,3,4,5
 1,2,3,4,5

第二个文件名为book2.csv，如下所示：

 header1,header2,header3,header4,header5
 1,2,3,4
 1,2,3,4
 1,2,3,4

我的目标是将book1.csv中包含5的列复制到book2.csv中相应的列。

我的代码的问题似乎是它没有正确地追加，也没有只选择要复制的索引。它还提供了一个错误，即我选择了一个不正确的索引位置。输出如下：

 header1,header2,header3,header4,header5
 1,2,3,4
 1,2,3,4
 1,2,3,41,2,3,4,5

这是我的代码：

 import csv

 with open('C:/Users/SAM/Desktop/book2.csv','a') as csvout:
    write=csv.writer(csvout, delimiter=',')
    with open('C:/Users/SAM/Desktop/book1.csv','rb') as csvfile1:
        read=csv.reader(csvfile1, delimiter=',')
        header=next(read)
        for row in read:
            row[5]=write.writerow(row)

我该怎么做才能正确地附加这个？

谢谢你的帮助！

Answer 1

尽管上面的一些代码可以工作，但它并不是真正的可伸缩性，需要一种矢量化的方法。与numpy或熊猫一起工作会使这些任务变得更容易，所以学习一点是很好的。

你可以从Pandas Website下载熊猫

# Load Pandas
from pandas import DataFrame

# Load each file into a pandas dataframe, this is based on a numpy array
data1 = DataFrame.from_csv('csv1.csv',sep=',',parse_dates=False)
data2 = DataFrame.from_csv('csv2.csv',sep=',',parse_dates=False)

#Now add 'header5' from data1 to data2
data2['header5'] = data1['header5']

#Save it back to csv
data2.to_csv('output.csv')

Answer 2

像这样的怎么样。我在这两本书中都读过，将book1的最后一个元素附加到book2行中，用于存储在列表中的book2中的每一行。然后我将该列表的内容写入一个新的.csv文件。

with open('book1.csv', 'r') as book1:
    with open('book2.csv', 'r') as book2:
        reader1 = csv.reader(book1, delimiter=',')
        reader2 = csv.reader(book2, delimiter=',')

        both = []
        fields = reader1.next() # read header row
        reader2.next() # read and ignore header row
        for row1, row2 in zip(reader1, reader2):
            row2.append(row1[-1])
            both.append(row2)

        with open('output.csv', 'w') as output:
            writer = csv.writer(output, delimiter=',')
            writer.writerow(fields) # write a header row
            writer.writerows(both)

Answer 3

关于“我选择了不正确的索引位置的错误”，我怀疑这是因为您在代码中使用了row[5]。Python中的索引从0开始，因此如果有A = [1, 2, 3, 4, 5]，那么要得到5，就需要print(A[4])。

假设这两个文件的行数相同，并且行的顺序相同，我想您应该这样做：

import csv
# Open the two input files, which I've renamed to be more descriptive,
# and also an output file that we'll be creating
with open("four_col.csv", mode='r') as four_col, \
     open("five_col.csv", mode='r') as five_col, \
     open("five_output.csv", mode='w', newline='') as outfile:
  four_reader = csv.reader(four_col)
  five_reader = csv.reader(five_col)
  five_writer = csv.writer(outfile)
  _ = next(four_reader) # Ignore headers for the 4-column file
  headers = next(five_reader)
  five_writer.writerow(headers)
  for four_row, five_row in zip(four_reader, five_reader):
    last_col = five_row[-1] # # Or use five_row[4]
    four_row.append(last_col)
    five_writer.writerow(four_row)