Python數據展示 - 生成表格圖片 - ⎝⎛CodingNote.cc ⎞⎠

Python數據展示 – 生成表格圖片

2022 年 4 月 9 日
筆記
Python

前言

前一篇文章介紹了推送信息到企業微信群里，其中一個項目推送的信息是使用Python自動生成的表格，本文來講講如何用Python生成表格圖片。

選一個合適庫

Python最大的優點就是第三方庫豐富，基本你要什麼功能，都能找到別人實現好的庫，幾行代碼一調用就完事了。

Pytable

項目地址：//github.com/HiroshiARAKI/pytable

最先找到的是日本人開發的pytab庫，它是基於matplotlib來畫圖的，默認參數下生成的表格外觀一般般，而且顯示不了中文字符，字體也很小，效果不理想。

我一開始還抱着死磕的心態，把這個庫的源碼下載下來魔改，改了字體和大小，生成的表格終於能看了，但排版還是會出各種奇奇怪怪的問題，比如文字溢出單元格之類的，心累……

Plotly

官網地址：//plotly.com/

後面找到了一個新的庫：plotly，官網的介紹是：

The front end for ML and data science models

專門為機器學習和數據科學設計的前端展示工具，單純拿來畫表格還算大材小用了~ 它是用網頁來渲染的，看例子效果還可以，我就換成這個plotly試試，嗯，真的可以，那就這個了。

本文會分別介紹這兩個庫的使用，包括我魔改了pytab的地方。

（不過還是推薦plotly，網頁渲染就是好）

pytab

首先安裝

pip install pytab

然後我把畫表格的代碼封裝成一個函數

import uuid
from typing import List, Dict, Optional, Tuple

import pytab

def draw_table(data: Dict[str, list]):
    """
    畫表格

    :param data: 數據格式 {
        'a': [1.0, 2.1, 3.5, 4.0, 2.0, 1.0, 2.1, 3.5, 4.0, 2.0, ],
        'b': [5.7, 6.1, 7.2, 8.3, 1.2, 5.7, 6.1, 7.2, 8.3, 1.2, ],
        }
    :return:
    """
    # 設置字體，不然顯示不了中文
    pytab.plt.rcParams["font.sans-serif"] = ["SimHei"]
    pytab.table( 
        data=data,
        data_loc='center',
        # th_type='dark',
        th_c='#aaaaee',  # 設置表頭背景顏色
        td_c='gray',  # 設置數據行背景顏色
        table_type='striped',
        figsize=(len(data.keys()), int(len(data.values()) / len(data.keys()) + 1)),
        # fontsize=18,
    )

    # pytab.show()
    temp_file = os.path.join(tempfile.gettempdir(), f'{uuid.uuid4().hex}.jpg')
    print(temp_file)
    pytab.save(temp_file)
    return temp_file

按照注釋的這個數據格式，a和b是表頭列名，後面的數組是每一列的數據，很好理解

{
    'a': [1.0, 2.1, 3.5, 4.0, 2.0, 1.0, 2.1, 3.5, 4.0, 2.0, ],
    'b': [5.7, 6.1, 7.2, 8.3, 1.2, 5.7, 6.1, 7.2, 8.3, 1.2, ],
}

畫出來的表格是這樣的

再來試試中文顯示會怎麼樣

{
    '列1': [1.0, 2.1, 3.5, 4.0, 2.0, 1.0, 2.1, 3.5, 4.0, 2.0, ],
    '列2': [5.7, 6.1, 7.2, 8.3, 1.2, 5.7, 6.1, 7.2, 8.3, 1.2, ],
}

畫出來是這樣

有點丑吧，勉強能看

OK~ 關於pytab的就不多折騰了，畢竟上限就在這了

接下來看看plotly

plotly

首先安裝

pip install plotly

話不多說，老規矩，我又是寫成一個函數

import uuid
from typing import List, Dict, Optional, Tuple

import plotly.graph_objects as go
import plotly.io as pio

def draw_table(headers: List[str], cells: List[list]):
    """
    畫表

    :param headers: header=dict(values=['A Scores', 'B Scores'])
    :param cells: cells=dict(values=[[100, 90, 80, 90], [95, 85, 75, 95]])
    :return:
    """
    pio.kaleido.scope.default_width = len(','.join(headers)) * 20
    pio.kaleido.scope.default_height = 250 + len(cells[0]) * 20
    fig = go.Figure(data=[go.Table(header=dict(values=headers), cells=dict(values=cells))])
    # fig.show()
    image_file = os.path.join(tempfile.gettempdir(), f'{uuid.uuid4().hex}.jpg')
    print('write image to', image_file)
    fig.write_image(image_file)
    return image_file

這次的參數格式更前面的pytab不一樣，要傳兩個參數，都是數組類型

第一個是表頭，第二個數組是單元格

先來個例子試試

draw_table(['列A', '列B'], [[100, 90, 80, 90], [95, 85, 75, 95]])

然後把上面函數代碼里的fig.show()注釋去掉，可以看到生成的表格圖片效果

運行後可以發現自動打開了瀏覽器，因為這個庫是使用網頁來渲染表格的，效果如下

效果比前面那個pytab好一些，哈哈~

OK，就這樣啦。

下次分享一些關於最近用Django開發後台的。

參考資料

python用prettytable輸出漂亮的表格：//linuxops.org/blog/python/prettytable.html
Plotly關於表格的文檔：//plotly.com/python/table/
Plotly關於導出圖片的文檔：//plotly.com/python/static-image-export/
python將表格展示的數據生成圖片：//www.jianshu.com/p/3bfbf1bf0699

Tags: Python