python : How to create a 2D density map/heat map

Question

我正在用 python 编码。我有 3 个数组 x、y 和 z，我想用 colorbar 绘制平面 (x,y) 中 z 值的二维密度图。

所以在我的图中，点 x[0] 和 y[0] 的颜色将由 z[0] 的值决定，点 x[1] 和 y[1] 的颜色将是由z[1]等的值决定

有人知道怎么做吗？

谢谢

Answer 1

查看 https://matplotlib.org/api/_as_gen/matplotlib.pyplot.scatter.html

对于不同的颜色图:https://matplotlib.org/tutorials/colors/colormaps.html

您需要的一段代码示例如下所示

 #--------------------------Plotting starts here---------------------------------#

    fig, ax0 = plt.subplots()

    im0 = plt.scatter(x,y,s=1,c=z, cmap='bwr')


#------------------if you want to use pcolormesh-------------------
#----------and have Z values stored as a numpy array Data---------------------#

    #X,Y = np.meshgrid(x,y)
    #im0 = ax0.pcolormesh(X,Y,Data, cmap="YourFavouriteColormap')

    cbar = fig.colorbar(im0,ax=ax0)
    ax0.set_title("Your title")
    plt.xlabel("xlabel")
    plt.ylabel("ylabel")
    filename = "prefix" + "."+ "fileformat"
    plt.savefig(filename)

编辑 1:从您的一条评论中，如果您有网格数据，可以尝试 pcolormesh并尝试着色，这是插值的可选参数。

 shading{'flat', 'gouraud'}, optional

    The fill style, Possible values:

        'flat': A solid color is used for each quad. The color of the quad (i, j), (i+1, j), (i, j+1), (i+1, j+1) is given by C[i, j].
        'gouraud': Each quad will be Gouraud shaded: The color of the corners (i', j') are given by C[i',j']. The color values of the area in between is interpolated from the corner values. When Gouraud shading is used, edgecolors is ignored.