Python 更新matplotlib热图的美学 导入matplotlib.pyplot作为plt 将numpy作为np导入 列标签=列表（'ABCD'） 行标签=列表（'WXYZ'） 数据=np.random.rand（4,4） 图，ax=plt.子批次（） 热图=ax.pcolor（数据，cmap=plt.cm.blue） #将主刻度放在每个单元格的中间 ax.set_xticks（np.arange（data.shape[0]）+0.5，minor=False） ax.set_yticks（np.arange（data.shape[1]）+0.5，minor=False） #想要更自然、更像桌子的展示吗 ax.invert_yaxis（） ax.xaxis.set_label_position（'top'）#

我想你需要两个技巧。首先，添加行

import matplotlib.pyplot as plt
import numpy as np
column_labels = list('ABCD')
row_labels = list('WXYZ')
data = np.random.rand(4,4)
fig, ax = plt.subplots()
heatmap = ax.pcolor(data, cmap=plt.cm.Blues)

# put the major ticks at the middle of each cell
ax.set_xticks(np.arange(data.shape[0])+0.5, minor=False)
ax.set_yticks(np.arange(data.shape[1])+0.5, minor=False)

# want a more natural, table-like display
ax.invert_yaxis()
ax.xaxis.set_label_position('top') # <-- This doesn't work!

ax.set_xticklabels(row_labels, minor=False)
ax.set_yticklabels(column_labels, minor=False)
plt.show()

使单元格显示为正方形（假设x轴和y轴上的数字相等，如示例中所示）。如果x轴上有x个正方形，y轴上有y个正方形，我想你可以这样做

ax.set_aspect('equal')

其次，您需要将edgecolor添加到单元格中，并使边缘变厚，因此请修改线条，例如

ax.set_aspect(float(y) / float(x))

结果是

---

您想要单元格之间的空格吗？是的，还有方形单元格，而且我希望单元格大小可以指定为tanks@innsfree，这看起来很棒！但是，我的数据集可能没有相同数量的行和列。添加了关于该情况的注释

heatmap = ax.pcolor(data, cmap=plt.cm.Blues, edgecolor='white', linewidths=10)