当set_position和set_size_inches在matplotlib中覆盖、嵌入和散布时不能正常工作

Question

我试图在np.meshgrid(grid_x, grid_y)定义的光栅上从矩阵np.meshgrid(grid_x, grid_y)创建一个图像，这样vx=grid_x[i], vy=grid_y[j]图像的值就是z2[i, j]。在这张图像的上方，我试图添加三个向量x, y, z获得的一些点的散点图，使第一个点具有坐标(x[k], y[k])和值z[k]。所有这些零散点都在上述光栅区域内。

下面是我试图绘制的上述数据的一个例子。

import numpy as np

np.random.seed(1)
z2 = np.ones((1000, 1000)) * 0.66
z2[0, 0] = 0
z2[-1, -1] = 1
x = np.random.rand(1000) * 1000
y = np.random.rand(1000) * 1000
z = np.random.rand(1000)
grid_x = np.linspace(0, 999, 1000)
grid_y = np.linspace(0, 999, 1000)

为了做到这一点，我使用了一个2D图，其中x和y值被用来定义点的位置，z由从彩色地图绘制的颜色表示。

这幅图像的要求是: 1)在实际的图形和图形的边缘之间不应该有空白；2) x和y轴上的单位长度应该相等；3)图像不应该太大。为了实现这些，我使用下面的代码进行绘图。

import matplotlib.pyplot as plt
from matplotlib import cm

def plot_img(x, y, z, grid_x, grid_y, z2, set_fig_size=True):

    # determine the figure size
    if set_fig_size:
        height, width = np.array(z2.shape, dtype=float)
        dpi = max(max(640 // height, 640 // width), 1)
        width, height = width * dpi, height * dpi
        plt.gcf().set_size_inches(width, height)
        plt.gcf().set_dpi(dpi)

    # plot the figure
    plt.gca().axis('off')
    plt.gca().axis('equal')
    plt.gca().set_position([0, 0, 1, 1])
    plt.xlim((grid_x[0], grid_x[-1]))
    plt.ylim((grid_y[0], grid_y[-1]))

    # the raster
    cmap = cm.get_cmap('gray')
    cmap.set_bad(color='red', alpha=0.5)
    plt.imshow(z2, cmap=cmap, interpolation='none', origin='lower',
                   extent=(grid_x[0], grid_x[-1], grid_y[0], grid_y[-1]))

    # the scatter plot
    min_z, max_z = np.min(z), np.max(z)
    c = (z - min_z) / (max_z - min_z)
    plt.scatter(x, y, marker='o', c=c, cmap='Greens')

    plt.show()

奇怪的是，当我使用上述示例数据运行plot_img(x, y, z, grid_x, grid_y, z2)时，会显示以下图像。

​

​

本质上，只有栅格数据被绘制，而散乱的数据不是。

然后我尝试了plot_img(x, y, z, grid_x, grid_y, z2, set_fig_size=False)。结果是

​

​

请注意，为了清楚地显示图中的空白，我保留了围绕它的PyCharm背景。基本上，我不希望在这个数字中包括一些空白。

我想知道为什么会发生这种情况，以及如何修复代码以获得正确的输出，这基本上是没有空格的第二个结果。谢谢!

Answer 1

将dpi和figsize代码替换为

# determine the figure size
height, width = np.array(z2.shape, dtype=float)
dpi = 200
# get size in inches:
width, height = height / dpi, width / dpi
plt.gcf().set_size_inches(width, height)
plt.gcf().set_dpi(dpi)

你将有一个1000x1000像素的数字，在200 dpi是5"x5“。