python写入excel文件

迷南。 2023-07-03 11:00 25阅读 0赞

# coding:utf-8
    import xlrd
    import xlwt
    
    
    # 设置表格样式
    def set_style(name, height, bold=False):
        style = xlwt.XFStyle()
        font = xlwt.Font()
        font.name = name
        font.bold = bold
        font.color_index = 4
        font.height = height
        style.font = font
        return style
    
    
    def add_sheet(excel_obj, sheet_name):
        """
        增加sheet页
        :param excel_obj: 创建的excel文件对象
        :param sheet_name: sheet名称
        :return: new_sheet 返回新创建的sheet对象
        """
        # cell_overwrite_ok=True表示一个单元格可以被多次覆盖写入
        new_sheet = excel_obj.add_sheet(sheet_name, cell_overwrite_ok=True)
        return new_sheet
    
    
    def add_sheet_row(sheet_obj, fieldnames, row_index=0):
        """
        在sheet中写入一行内容
        :param sheet_obj: sheet对象
        :param fieldnames: 要写入的内容列表，列表中每个元素写入一个单元格
        :param row_index: 要写入的行号，默认写入第0行
        :return:
        """
        for i in range(len(fieldnames)):
            # 在第0行，第i列，以set_style设置的格式，写入fieldnames[i]中的内容
            sheet_obj.write(row_index, i, fieldnames[i], set_style('Times New Roman', 220, True))
    
    
    def add_sheet_col(sheet_obj, fieldnames, col_index=0, row_index=1):
        """
        在sheet的某一列中写入内容
        :param sheet_obj: sheet对象
        :param fieldnames: 要写入的内容列表，列表中每个元素写入一个单元格
        :param col_index: 要写入的列编号，默认滴入第0列
        :param row_index: 从row_index行开始写入，默认从第一行开始写，第0行是标题
        :return:
        """
        for i in range(len(fieldnames)):
            sheet_obj.write(row_index + i, col_index, fieldnames[i], set_style('Times New Roman', 220, True))
    
    
    def add_sheet_cell(sheet_obj, cell, row_index, col_index):
        """
        在sheet的某个单元格中写入内容
        :param sheet_obj: sheet对象
        :param cell: 待写入的内容
        :param row_index: 单元格的行号
        :param col_index: 单元格的列号
        :return:
        """
        sheet_obj.write(row_index, col_index, cell)
    
    
    def merge_cell(sheet_obj, row1_index, row2_index, col1_index, col2_index, cell):
        """
        合并单元格
        :param sheet_obj:sheet对象
        :param row1_index:起始单元格的行号
        :param row2_index:截止单元格的行号
        :param col1_index:起始单元格的列号
        :param col2_index:截止单元格的列号
        :param cell: 单元格合并后写入的内容
        :return:
        """
        sheet_obj.write_merge(row1_index, row2_index, col1_index, col2_index, label=cell)
    
    
    # 写Excel
    def write_excel():
        f = xlwt.Workbook()
        sheet1 = add_sheet(f, '学生')
        row0 = ["姓名", "年龄", "出生日期", "爱好"]
        colum0 = ["张三", "李四", "恋习Python", "小明", "小红", "无名"]
    
        # 写入一行(属性列，如姓名、年龄、性别)
        add_sheet_row(sheet1, row0)
        # 写入一列
        add_sheet_col(sheet1, colum0, 0, 1)
        # 写入一个单元格
        add_sheet_cell(sheet1, '2006/12/12', 1, 2)
    
        # 合并单元格
        merge_cell(sheet1, 6, 6, 1, 3, '未知')    # 合并行单元格，合并第6行的1/2/3三个单元格
        merge_cell(sheet1, 1, 2, 3, 3, '打游戏')   # 合并列单元格，合并第1/2行的第3个单元格
        merge_cell(sheet1, 4, 5, 3, 3, '打篮球')
    
        f.save('./data/test_write.xls')
    
    
    if __name__ == '__main__':
        write_excel()