Question

这就是问题，我有4个.csv文件，我想从这4个文件中的每个文件中复制一列，并将它们合并为一个单独的列作为新的.csv主文件的第一列。然后，使用另一组不同的列再次重复相同的过程，并将其存储为主文件的第二列，并再次执行至少12次。

到目前为止，这是我的代码：

CODE -

import os
import csv
import datetime as dt
from os import listdir
from os.path import join 
import matplotlib.pyplot as plt



#get the list of files in mypath and store in a list

mypath = 'C:/Users/Alan Cedeno/Desktop/Test_Folder/'
onlycsv = [f for f in listdir(mypath) if '.csv' in f]

#print out all the files with it's corresponding index

for i in range(len(onlycsv)):
print(i,onlycsv[i])

#prompt the user to select the files

option = input('please select file1 by number: ')
option2 = input('please select file2 by number: ')

#build out the full paths of the files and open them

fullpath1 = join(mypath, onlycsv[option])
fullpath2 = join(mypath, onlycsv[option2])

#create third new.csv file

root, ext = os.path.splitext(fullpath2)
output = root + '-new.csv'

with open(fullpath1) as r1, open(fullpath2) as r2, open(output, 'a') as w:
    writer = csv.writer(w)
    merge_from = csv.reader(r1)
    merge_to = csv.reader(r2)
    # skip 3 lines of headers
    for _ in range(3):
    next(merge_from)
    for _ in range(1):
        next(merge_to)
    for merge_from_row, merge_to_row in zip(merge_from, merge_to):
        # insert from col 0 as to col 0
        merge_to_row.insert(1, merge_from_row[2])
        # replace from col 1 with to col 3
        #merge_to_row[0] = merge_from_row[2]
        # delete merge_to rows 5,6,7 completely
        #del merge_to_row[5:8]
        writer.writerow(merge_to_row)

有什么建议吗？

如果帖子需要格式化，请告诉我。

Answer 1

你想在这做什么？如果你想将所有csv合并为一个，那么它就会很好。

with open(fullpath1) as r1, open(fullpath2) as r2, open(output, 'a') as fout:
   for line in r1:
      fout.write(line)
   for line in r2:
      fout.write(line)
   fout.close()

Python-从多个.csv文件中获取列并将其粘贴到新的.csv主文件中

1 个答案: