如何使用matplotlib和numpy绘制热力图

Question

现有3个list

x = [x0, x1, x2, ..., xn]
y = [y0, y1, y2, ..., yn]
v = [v0, v1, v2, ..., vn]

希望可以据此生成热力图，效果类似http://stackoverflow.com/questions/21663358/gnuplot-heatmap-xyz 答案里的效果。搜索了一些类似的问题但是好像不太一样(http://stackoverflow.com/questions/14391959/heatmap-in-matplotlib-with-pcolor).

谢谢！

Answer 1

height = np.max(y) + 1
width = np.max(x) + 1
arr = np.zeros((height, width))
for i in range(len(x)):
    arr[y[i], x[i]] = v[i]

plt.matshow(arr, cmap='hot')
plt.colorbar()
plt.show()

此外还可以换颜色，执行dir(plt.cm)来查看可用的颜色表。

Answer 2

或许你能参考下面链接的内容

http://discuss.analyticsvidhya.com/t/how-to-plot-heat-map-in-python/1922/2

另外不限定matplotlib的話，可以嘗試 Plotly

https://plot.ly/python/heatmaps/