checkerboard_plot：在 matplotlib 中创建棋盘格图

通过 matplotlib 绘制棋盘格图/热力图的函数

from mlxtend.plotting import checkerboard_plot

概述

通过 matplotlib 绘制棋盘格图/热力图的函数。

参考文献

  • -

示例 1 - 默认

from mlxtend.plotting import checkerboard_plot
import matplotlib.pyplot as plt
import numpy as np

ary = np.random.random((5, 4))

brd = checkerboard_plot(ary)
plt.show()

png

示例 2 - 更改颜色和标签

from mlxtend.plotting import checkerboard_plot
import matplotlib.pyplot as plt
import numpy as np

checkerboard_plot(ary, 
                  col_labels=['abc', 'def', 'ghi', 'jkl'],
                  row_labels=['sample %d' % i for i in range(1, 6)],
                  cell_colors=['skyblue', 'whitesmoke'],
                  font_colors=['black', 'black'],
                  figsize=(4.5, 5))
plt.show()

png

API

checkerboard_plot(ary, cell_colors=('white', 'black'), font_colors=('black', 'white'), fmt='%.1f', figsize=None, row_labels=None, col_labels=None, fontsize=None)

通过 matplotlib 绘制棋盘格表/热力图。

参数

  • ary : 类数组对象，形状 = [n, m]

    一个二维 NumPy 数组。

  • cell_colors : 元组或列表（默认值：('white', 'black')）

    包含棋盘格模式的两种颜色的元组或列表。

  • font_colors : 元组或列表（默认值：('black', 'white')）

    与单元格颜色对应的字体颜色。

  • figsize : 元组（默认值：(2.5, 2.5)）

    图形的高度和宽度

  • fmt : 字符串（默认值：'%.1f'）

    用于单元格值的 Python 字符串格式化程序。默认值 '%.1f' 将产生小数点后一位的浮点数。使用 '%d' 将数字显示为整数。

  • row_labels : 列表（默认值：None）

    行标签列表。默认使用数组的行索引 0 到 n。

  • col_labels : 列表（默认值：None）

    列标签列表。默认使用数组的列索引 0 到 m。

  • fontsize : 整数（默认值：None）

    指定棋盘格表的字体大小。如果为 None，则使用 matplotlib 的默认值。

返回值

  • fig : matplotlib Figure 对象。

示例

有关使用示例，请参阅 https://mlxtend.cn/mlxtend/user_guide/plotting/checkerboard_plot/