python從數據庫取數據后寫入excel 使用pandas.ExcelWriter設置單元格格式

用python從數據庫中取到數據后，寫入excel中做成自動報表，ExcelWrite默認的格式一般來說都比較丑，但workbook提供可以設置自定義格式，簡單記錄個demo，供初次使用者參考。

一、 取幾列數據長下面這樣，數字我隨便編的，不用在意：

 

二、ExcelWriter寫入后默認格式長下面這樣，字體丑，寬度不夠，沒有百分比格式：

 

三、設置格式后保存結果，好看多了：

 

四、好了，實現在下面代碼里有注釋

# -*- coding: utf-8 -*-

import pandas as pd
import sys

reload(sys)
sys.setdefaultencoding('utf8')
from mlutil.database import MySQLDB
from datetime import datetime, timedelta


def report():
    # 1.取數據
    m = MySQLDB('/data/wowo/password.p', 'mysql')
    m.connect()
    df1 = pd.DataFrame(m.query('select * from table1'))
    df1 = df1[[u'app_date',
               u'cust_cnt',
               u'app_cnt',
               u'overdue_amt',
               u'overdue_rate',
               ]]
    df1.sort_values(by='app_date', ascending=False, inplace=True)
    df1 = df1.set_index(u'app_date')
    df1.loc[u'全部'] = df1.apply(lambda x: x.sum())

    df1.columns = [u'人數',
                   u'交易數',
                   u'逾期金額',
                   u'逾期率']
    df1 = df1.reset_index()

    # 2.保存至excel文件
    t = datetime.now().date() - timedelta(days=1)
    writer = pd.ExcelWriter(
        u'/home/wowo/daily_report/測試_%d%02d%02d.xlsx' % (t.year, t.month, t.day))
    workbook = writer.book

    # 3.設置格式
    fmt = workbook.add_format({"font_name": u"微軟雅黑"})
    percent_fmt = workbook.add_format({'num_format': '0.00%'})
    amt_fmt = workbook.add_format({'num_format': '#,##0'})
    border_format = workbook.add_format({'border': 1})
    note_fmt = workbook.add_format(
        {'bold': True, 'font_name': u'微軟雅黑', 'font_color': 'red', 'align': 'left', 'valign': 'vcenter'})
    date_fmt = workbook.add_format({'bold': False, 'font_name': u'微軟雅黑', 'num_format': 'yyyy-mm-dd'})

    date_fmt1 = workbook.add_format(
        {'bold': True, 'font_size': 10, 'font_name': u'微軟雅黑', 'num_format': 'yyyy-mm-dd', 'bg_color': '#9FC3D1',
         'valign': 'vcenter', 'align': 'center'})
    highlight_fmt = workbook.add_format({'bg_color': '#FFD7E2', 'num_format': '0.00%'})

    # 4.寫入excel
    l_end = len(df1.index) + 2
    df1.to_excel(writer, sheet_name=u'測試頁簽', encoding='utf8', header=False, index=False, startcol=0, startrow=2)
    worksheet1 = writer.sheets[u'測試頁簽']
    for col_num, value in enumerate(df1.columns.values):
        worksheet1.write(1, col_num, value, date_fmt1)

    # 5.生效單元格格式
    # 增加個表格說明
    worksheet1.merge_range('A1:B1', u'測試情況統計表', note_fmt)
    # 設置列寬
    worksheet1.set_column('A:E', 15, fmt)
    # 有條件設定表格格式：金額列
    worksheet1.conditional_format('B3:E%d' % l_end, {'type': 'cell', 'criteria': '>=', 'value': 1, 'format': amt_fmt})
    # 有條件設定表格格式：百分比
    worksheet1.conditional_format('E3:E%d' % l_end,
                                  {'type': 'cell', 'criteria': '<=', 'value': 0.1, 'format': percent_fmt})
    # 有條件設定表格格式：高亮百分比
    worksheet1.conditional_format('E3:E%d' % l_end,
                                  {'type': 'cell', 'criteria': '>', 'value': 0.1, 'format': highlight_fmt})
    # 加邊框
    worksheet1.conditional_format('A1:E%d' % l_end, {'type': 'no_blanks', 'format': border_format})
    # 設置日期格式
    worksheet1.conditional_format('A3:A62', {'type': 'no_blanks', 'format': date_fmt})

    # 6.保存
    writer.save()


if __name__ == '__main__':
    report()