matplotlib 如何使子地块的大小均匀

ars1skjm  于 2023-06-23  发布在  其他
关注(0)|答案(2)|浏览(136)

我使用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子图的示例

2w2cym1i

2w2cym1i1#

不要使用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()

结果图:

vdzxcuhz

vdzxcuhz2#

我假设您希望以不同的大小显示图像,这样不同图像的所有像素大小相同。
这通常是困难的,但是对于子图网格的行(或列)中的所有图像具有相同大小的情况,它变得容易。这个想法可以是使用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()

相关问题