如何平均分配子图的大小？

Question

我正在使用matplotlib和GridSpec在3x3子图中绘制9张图像。

    fig = plt.figure(figsize=(30,40))
    fig.patch.set_facecolor('white')
    gs1 = gridspec.GridSpec(3,3)
    gs1.update(wspace=0.05, hspace=0.05)
    ax1 = plt.subplot(gs1[0])
    ax2 = plt.subplot(gs1[1])
    ax3 = plt.subplot(gs1[2])
    ax4 = plt.subplot(gs1[3])
    ax5 = plt.subplot(gs1[4])
    ax6 = plt.subplot(gs1[5])
    ax7 = plt.subplot(gs1[6])
    ax8 = plt.subplot(gs1[7])
    ax9 = plt.subplot(gs1[8])
    ax1.imshow(img1,cmap='gray')
    ax2.imshow(img2,cmap='gray')
    ...
    ax9.imshow(img9,cmap='gray')

但是，图像的每一行都有不同的大小。例如，第一行的图片大小为256x256，第二行的图片大小为200x200，第三行的图片大小为128x128

我想在子图中以相同大小绘制图像。我应该如何在python中使用它？谢谢

这是4x3子图的示例

Answer 1

不要使用matplotlib.gridspec，而要使用figure.add_subplot，如下面的给定工作代码所示。但是，在进行某些绘图时，您需要set_autoscale_on(False)来抑制其大小调整行为。

import numpy as np
import matplotlib.pyplot as plt

# a function that creates image array for `imshow()`
def make_img(h):
    return np.random.randint(16, size=(h,h)) 

fig = plt.figure(figsize=(8, 12))
columns = 3
rows = 4
axs = []

for i in range(columns*rows):
    axs.append( fig.add_subplot(rows, columns, i+1) )

    # axs[-1] is the new axes, write its title as `axs[number]`
    axs[-1].set_title("axs[%d]" % (i))

    # plot raster image on this axes
    plt.imshow(make_img(i+1), cmap='viridis', alpha=(i+1.)/(rows*columns))

    # maniputate axs[-1] here, plot something on it
    axs[-1].set_autoscale_on(False)   # suppress auto sizing
    axs[-1].plot(np.random.randint(2*(i+1), size=(i+1)), color="red", linewidth=2.5)


fig.subplots_adjust(wspace=0.3, hspace=0.4)
plt.show()

结果图：

Answer 2

我想您要显示不同大小的图像，以使不同图像的所有像素大小均相等。

这通常很难，但是对于子图网格的行（或列）中的所有图像都具有相同大小的情况，这变得容易。这个想法可以是使用gridspec的height_ratios（或在列的情况下为width_ratios）参数，并将其设置为图像的像素高度（宽度）。

import matplotlib.pyplot as plt
import numpy as np

images = [np.random.rand(r,r) for r in [25,20,12] for _ in range(3)]


r = [im.shape[0] for im in images[::3]]
fig, axes = plt.subplots(3,3, gridspec_kw=dict(height_ratios=r, hspace=0.3))

for ax, im in zip(axes.flat, images):
    ax.imshow(im)


plt.show()