2

我有两个文件，第一个叫做 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)

我应该怎么做才能让它正确附加？

谢谢你的帮助！

4

4 回答 4

5

这样的事情怎么办。我在两本书中都读过，将 中的每一行的最后一个元素附加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)
于 2013-09-03T16:44:32.920 回答
2

尽管上面的一些代码可以工作，但它并不是真正可扩展的，需要矢量化方法。开始使用 numpy 或 pandas 将使其中一些任务更容易，因此学习一点它是很棒的。

您可以从Pandas 网站下载 pandas

# 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')
于 2013-09-04T10:21:45.417 回答
1

关于“我选择了不正确的索引位置的错误”，我怀疑这是因为您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)
于 2013-09-03T16:39:55.580 回答
0

为什么不逐行读取文件并使用 -1 索引查找最后一项？

endings=[]

with open('book1.csv') as book1:
   for line in book1:        
      # if not header line:  
      endings.append(line.split(',')[-1])

linecounter=0
with open('book2.csv') as book2:
   for line in book2:
      # if not header line:
      print line+','+str(endings[linecounter]) # or write to file
      linecounter+=1

如果行号不匹配，您还应该捕获错误。

于 2013-09-03T17:24:42.393 回答