2017-07-18 188 views
1

我使用下面的代碼將多個.csv文件中的數據合併到一個.xls文件中。使用此代碼,所有數據都保存在一張表中,但我需要將每個文件的數據保存在一張單獨的表格中,但保存在同一個文件中。如何使用Python將多個.csv文件合併到一個.xls文件中?

你能幫我解決這個問題嗎?

import glob 
import csv 

csvfiles=glob.glob('*.csv') 
wf=csv.writer(open('output.xls', 'wb')) 

for files in csvfiles: 
    rd=csv.reader(open(files, 'r')) 
    rd.next() 
    for row in rd: 
     wf.writerow(row) 
+0

你可能想看看熊貓。它具有非常強大的[數據I/O](https://pandas.pydata.org/pandas-docs/stable/api.html#flat-file)功能。 ['read_csv'](https://pandas.pydata.org/pandas-docs/stable/generated/pandas.read_csv.html#pandas.read_csv)和['to_excel'](https://pandas.pydata.org /pandas-docs/stable/generated/pandas.DataFrame.to_excel.html)都可以輕鬆完成您要求的內容。 – saintsfan342000

回答

1

我也陷入了編碼錯誤,所以我這裏使用的https://docs.python.org/2/library/csv.html發現Unicode的CSV閱讀器。這個答案適用於Python 2.如果你需要Python 3的答案,請告訴我。

import csv, codecs, cStringIO, glob, os 
import xlsxwriter 

class UTF8Recoder: 
    """Iterator that reads an encoded stream and reencodes the input to UTF-8""" 
    def __init__(self, f, encoding): 
     self.reader = codecs.getreader(encoding)(f) 
    def __iter__(self): 
     return self 
    def next(self): 
     return self.reader.next().encode("utf-8") 

class UnicodeReader: 
    """A CSV reader which will iterate over lines in the CSV file "f", 
    which is encoded in the given encoding.""" 
    def __init__(self, f, dialect=csv.excel, encoding="utf-8", **kwds): 
     f = UTF8Recoder(f, encoding) 
     self.reader = csv.reader(f, dialect=dialect, **kwds) 
    def next(self): 
     row = self.reader.next() 
     return [unicode(s, "utf-8") for s in row] 
    def __iter__(self): 
     return self 

wb = xlsxwriter.Workbook('output.xlsx') 
for csvfile in csvfiles: 
    ws = wb.add_worksheet(os.path.split(csvfile)[-1][:30]) 
    with open(csvfile,'rb') as f: 
     ur = UnicodeReader(f) 
     for r, row in enumerate(ur): 
      for c, col in enumerate(row): 
       ws.write(r, c, col) 
wb.close()