可视化: Python—MatPlotLib—二维表格

最新推荐文章于 2024-10-12 07:00:00 发布

无聊的人生事无聊

最新推荐文章于 2024-10-12 07:00:00 发布

阅读量5.1k

点赞数 2

分类专栏：信息科学文章标签：可视化

本文链接：https://blog.csdn.net/Wangpeiyi9979/article/details/99130691

版权

信息科学专栏收录该内容

159 篇文章 1 订阅

订阅专栏

文章目录

一、图示
二、代码
三、使用

一、图示

在这里插入图片描述

二、代码

from matplotlib import pyplot as plt
import numpy as np
from pandas import *

def plot_table(row, col, vals):
    """
    函数功能: 绘制二维表格，草图如下:
        -----------------
            |col1 |col2 |
        -----------------
        row1|value|value|
        -----------------
        row2|value|value|
        -----------------
    输入：
        row:string,(N)           #['row1', 'row2']
        col:string,(M)           #['col1', 'col2']
        vals:np, (N,M)          
    """
    
    R, C = len(row), len(col)
    idx = Index(row)
    df = DataFrame(np.random.randn(R, C), index=idx, columns=col)
    
    # 根据行数列数设置表格大小
    figC, figR = 2.25*C, R
    fig = plt.figure(figsize=(figC, figR))
    
    # 设置fig并去掉边框
    ax = fig.add_subplot(111, frameon=True, xticks=[], yticks=[])
    ax.spines['top'].set_visible(False)
    ax.spines['right'].set_visible(False)
    ax.spines['bottom'].set_visible(False)
    ax.spines['left'].set_visible(False)
    
    the_table=plt.table(cellText=vals, rowLabels=df.index, colLabels=df.columns, colWidths = [0.1]*vals.shape[1], rowLoc='center', loc='center',cellLoc='center')
    the_table.set_fontsize(20)
    
    # 伸缩表格大小常数
    the_table.scale(figR/R*2 ,figC/C*1.5)

三、使用

1、导入包

from matplotlib import pyplot as plt
import numpy as np
from pandas import *

2、构造参数

S =[
     [1,0.44,0.29,0.33],
     [0.44,1,0.35,0.32],
     [0.29,0.35,1,0.60],
     [0.33,0.32,0.60,1]
    ]

row = ["Chinese","English","Math","Physics"]
col = ["Chinese","English","Math","Physics"]
vals = np.array(S)

3、绘制