python - 使用 matplotlib 绘制热图

Question

我有一组三元组，每个三元组由(var1, var2, result)组成。9 个三元组列表的示例:

<type 'list'>: 
[(4, 0.7, 0.8530612244898, 0.016579670213527985), 
 (4, 0.6, 0.8730158730157779, 0.011562402525241757), 
 (6, 0.8, 0.8378684807257778, 0.018175985875060037), 
 (4, 0.8, 0.8605442176870833, 0.015586992159716321), 
 (2, 0.8, 0.8537414965986667, 0.0034013605443334316), 
 (2, 0.7, 0.843537414966, 0.006802721088333352), 
 (6, 0.6, 0.8480725623582223, 0.01696896774039503), 
 (2, 0.6, 0.84693877551025, 0.010204081632749995), 
 (6, 0.7, 0.8577097505669444, 0.019873637350220318)]

现在我想用它创建一个热图。Var1 可以是值 [2,4,6]，Var2 可以是 [0.6, 0.7, 0.8] 。共有 9 个结果。

这是我用来绘制热图的代码:

    # list of 3-tuples to 3 lists: x, y and weights
    # x (var1) = [2,4,6]
    # y (var2) = [0.6, 0.7, 0.8]
    # weights (res) = [....] (9 values)

    x, y = np.meshgrid(x, y)
    intensity = np.array(weights)

    plt.pcolormesh(x, y, intensity)
    plt.colorbar()  # need a colorbar to show the intensity scale
    plt.show()

这导致了这个图表:

我觉得奇怪的是只有 4 个部分，但我预计会有 9 个 (3x3)。有人可以阐明我在这里做错了什么吗？

Answer 1

meshgrid 正在创建九个点，而不是 9-patch 。
查看输出。您在 (2, 0.6) 处得到 1 点，在 (2, 0.7) 处得到 1 点，依此类推。