import xlwt
import os


class TxtToExcel(object):
    def __init__(self, path):
        self.path = path
        self.workbook = xlwt.Workbook()
        self.worksheet = self.workbook.add_sheet('mybook', cell_overwrite_ok=True)

    def read_content(self, file):
        with open(file, 'r', encoding='utf-8') as f:
            return f.readlines()

    def file_list(self):
        return os.listdir(self.path)

    def write_to_excel(self, row, content):
        content_list = content.split(" ")
        for con in content_list:
            self.worksheet.write(row, content_list.index(con), con)
            # self.workbook.save('a.xls')

    def get_file_path_list(self):
        file_path_list = [os.path.join(self.path, file) for file in self.file_list()]
        # content = []
        # for file in file_path_list:
        #     content.extend(self.read_content(file))

        gen = (self.read_content(file) for file in file_path_list)
        index = 0
        for g in gen:
            for i in g:
                self.write_to_excel(index + 1, i)
                index += 1
            self.workbook.save('a.xls')

        # for c in content:
        #     self.write_to_excel(content.index(c)+1, c)


if __name__ == '__main__':
    path = r"E:\qqq\test_excel"
    TxtToExcel(path).get_file_path_list()

更多推荐

python读取txt文件内容写入到excel表格中