如何使用 Python 将 .csv 文件合并为多个工作表中的一个 .xls 文件？

Question

我使用以下代码将多个 .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)

Answer 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()

Answer 2

或者你可以：

workbook = Workbook('../your_path/joined.xlsx')
counter = 0
for csv_file in glob.glob(os.path.join('../your_path/', '*.csv')):
    sheet_name = 'Sheet_' + str(counter)
    counter += 1
    worksheet = workbook.add_worksheet(sheet_name)
    with open(csv_file, 'rt', encoding='utf8') as f:
        reader = csv.reader(f)
        for r, row in enumerate(reader):
            for c, col in enumerate(row):
                worksheet.write(r, c, col)
workbook.close()