[英]Python heatmap plot colorbar
到目前为止,我正在研究一个看起来像这样的情节:
我正在使用以下代码:
import matplotlib.pyplot as plt
import numpy as np
column_labels = ['A','B']
row_labels = ['A','B']
data = np.random.rand(2,2)
print(type(data))
data = np.array([[1, 0.232], [0.119, 1]])
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.tick_top()
ax.set_xticklabels(row_labels, minor=False)
ax.set_yticklabels(column_labels, minor=False)
plt.show()
我想要得到的是这样的东西:
我可以在其中看到颜色条以及它们上的单元格的值。 我需要在代码中添加些什么才能获得这些?
谢谢
有关颜色栏,请参见以下示例 。 您猜对了,正确的函数调用只是ax.colorbar()
。
关于pcolor
图上的值,恐怕您必须使用ax.text()
“手动”添加它们。 计算并循环遍历单元格中心值,然后调用text()
函数。 您可以使用关键字参数horizontalalignment='center'
和verticalalignment='center'
来使文本值围绕x, y
居中。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.