问题描述
我有两个ndarray:Mat,labels 当前,我显示Mat:
plt.imshow(Mat,cmap='gray',vmin=0,vmax=1,interpolation='None')
标签具有与Mat相同的形状,并且lables [(i,j)]包含Mat [[i,j)]的标签。 如何在每个像素上显示标签?
解决方法
最简单的方法是使用Seaborn的heatmap。当annot=True
时,它将数据值打印到单元格中。但是annot=
也可以是标签矩阵。在这种情况下,将打印格式设置为字符串(fmt='s'
)很重要。 annot_kws=
可以设置其他关键字,例如字体大小或颜色。 x
和yticklabels
可以合并到对heatmap()
的调用中,也可以在以后使用matplotlib进行设置。
默认着色的一个重要好处是Sorn在浅色单元格上使用黑色,在深色单元格上使用白色。
这是一个使用一些utf8字符作为标签的示例。
from matplotlib import pyplot as plt
import numpy as np
import seaborn as sns
M,N = 5,10
mat = np.random.rand(M,N)
labels = np.random.choice(['X','☀','★','♛'],size=(M,N))
ax = sns.heatmap(mat,cmap="inferno",annot=labels,annot_kws={'fontsize': 16},fmt='s')
plt.show()
PS:文档中有matplotlib example,用于在没有Seaborn的情况下创建类似内容。它可以很容易地适应从不同矩阵打印字符串的情况,还可以添加测试以根据单元的暗度来更改颜色。