使用pcolor在matplotlib中进行热图绘制?

我想制作一个像这样的热图(显示在FlowingData上): 

http://img1.mukewang.com/5d9826c70001dfd205450861.jpg

源数据在这里,但是可以使用随机数据和标签,即


import numpy

column_labels = list('ABCD')

row_labels = list('WXYZ')

data = numpy.random.rand(4,4)

在matplotlib中制作热图非常简单:


from matplotlib import pyplot as plt

heatmap = plt.pcolor(data)

我什至发现了一个看起来正确的colormap参数:heatmap = plt.pcolor(data, cmap=matplotlib.cm.Blues)


但是除此之外,我不知道如何显示列和行的标签以及如何以正确的方向显示数据(起源在左上角而不是左下角)。


尝试操作heatmap.axes(例如heatmap.axes.set_xticklabels = column_labels)都失败了。我在这里想念什么?

慕码人2483693
浏览 1963回答 3
3回答
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python