Python实现Excel转换工具小结

经历过的打表工具从c++、C#，再到Python，算下来还是Python方便些。一天即可上手开发，非常适合快速迭代中的各种小工具开发。 Python开源的第三方库很多，涉及excel方面的也有好几个xlrd、xlwt、xlutils 等等。

1. xlrd\xlwt

xlrd 主要实现excel文件的读操作，xlwt主要负责写操作。xlutils 是将二者融合在一起。

安装

现在之后可以通过源码的形式进行安装，也可以通过exe的形式安装。对于只用报表格小工具的同学，可以通过批处理进行。

批处理代码：

msiexec /package python-2.7.10.64bit.msi /quiet

cd .\xlrd-0.9.4\

c:\Python27\python.exe setup.py install

cd ..\.\xlwt-1.0.0\

c:\Python27\python.exe setup.py install

cd ..\.\protobuf-2.6.1\python\

c:\Python27\python.exe setup.py install

pause

2. 操作实现

表格构成

Workbook ：一个完整的excel表格
Worksheet ：表格中一个sheet
Cell ：表中一个单元格

读操作

流程：读取book，获取sheet，然后根据col\row获得对应的数值，具体的API的可以参考：

http://xlrd.readthedocs.io/en/latest/api.html

实例（无容错处理）：

#打开profiles.xls

self.book = xlrd.open_workbook(from_this_dir('profiles.xls'), formatting_info=True)

#通过名字获得

self.sheet = self.book.sheet_by_name('PROFILEDEF')

#通过索引获得

self.sheet = data.sheets()[0]

self.sheet = data.sheet_by_index(0) #通过索引顺序获取

#获取整行和整列的值（数组）

self.sheet.row_values(i)

self.sheet.col_values(i)

#获取元素

cell = self.sheet.cell(1, 1)

写操作

流程：创建一个book，添加一个sheet并向对应Cell中添加数据。具体Api可以参

考：http://xlwt.readthedocs.io/en/latest/api.html

案例：

#创建book

book = xlwt.Workbook()

#添加sheet

sheet = book.add_sheet(sheet_name)

#写数据

sheet.write(0, 0, label = 'Row 0, Column 0 Value')

#保存book

book.save(file_name)

3.实现表格合并工具

#加载表格

class SheetLoader:

    # Excel 表加载类，用于加载Excel表

    def __init__(self):

        # 构造函数

        self.file_path = ''

        self.sheet_name = ''

        self.workbook = None

        self.sheet = None

        self.num_rows = 0

        self.num_cols = 0

    def load(self, file_path, sheet_name):

        # 解析Excel表格，成功返回True，否则返回False

        self.file_path = file_path

        self.sheet_name = sheet_name

        # 打开Excel文件

        try:

            self.workbook = xlrd.open_workbook(self.file_path)

        except BaseException, e:

            print 'Error: Load excel file failed', self.file_path

            sys.exit(-1)

        # 打开Excel表格

        try:

            self.sheet = self.workbook.sheet_by_name(self.sheet_name)

        except BaseException, e:

            print 'Load excel sheet failed', self.sheet_name

            sys.exit(-1)

        # 行数和列数

        self.num_rows = len(self.sheet.col_values(0))

        self.num_cols = len(self.sheet.row_values(0))

        print 'Load', self.file_path, self.sheet_name, self.num_rows, self.num_cols

        return True

class SheetCombiner:

    '''

    表格合并工具

    用于将多个表格合并成一个表格，所有表格必须有相同的行数

    '''

    def __init__(self):

        # 构造函数

        self.loaders = []

        self.num_cols = 0

    def add_sheet(self, file_path, sheet_name):

        '''

        添加表格

        :param file_path: Excel文件路径

        :param sheet_name: Excel表格名称

        :return:

        '''

        loader = SheetLoader()

        if not loader.load(file_path, sheet_name):

            print 'Error: Load excel file failed', self.file_path

            sys.exit(-1)

        if self.num_cols == 0:

            self.num_cols = loader.get_num_cols()

        else:

            # 检查表的列数是否一致

            if self.num_cols != loader.get_num_cols():

                print 'Error: not same cols', loader.get_sheet_name(), self.num_cols, loader.get_num_cols()

                sys.exit(-1)

            # 检查表的表头是否一致

            first = self.loaders[0]

            for r in range(3):

                for c in range(self.num_cols):

                    if loader.cell_value(r, c) != first.cell_value(r, c):

                        print 'Error: not same header', loader.get_sheet_name(), r, c

                        sys.exit(-1)

        self.loaders.append(loader)

    def combine(self, file_path, sheet_name):

        '''

        合并表格

        :param file_path: 合并的Excel文件名称

        :param sheet_name: 合并的Excel表格名称

        :return:

        '''

        if len(self.loaders) == 0:

            return

        workbook = xlwt.Workbook()

        sheet = workbook.add_sheet(sheet_name)

        loader = self.loaders[0]

        for row in range(4):

            for col in range(self.num_cols):

                sheet.write(row, col, loader.cell_value(row, col))

        row = 4

        for loader in self.loaders:

            for r in range(4, loader.get_num_rows()):

                for c in range(self.num_cols):

                    sheet.write(row, c, loader.cell_value(r, c))

                row += 1

        workbook.save(file_path)

实现表格合并：

combiner = SheetCombiner()

combiner.add_sheet('../dev/data_xls/achievement_data.xls', 'ACHIEVEMENT_DATA')

combiner.add_sheet('../dev/data_xls/task_data.xls', 'TASK_DATA')

combiner.combine('./output.xls', 'OUTPUT')

4. 结论

相对于lua这种脚本语言，python的功能确实强悍很多，尤其是众多的第三方库。以后可以愉快的考虑用python尝试一下各种小工具了。

参考：http://www.python-excel.org/

巴特西