在matplotlib中創建多個大小不等的列和行

Question

我需要在matplotlib中創建多個大小不等的列和行。 這是一個示例代碼：

a = np.random.rand(20, 20)
b = np.random.rand(20, 5)
c = np.random.rand(5, 20)
d = np.random.rand(5,5)
arrays = [a,b,c,d]
fig, axs = plt.subplots(2, 2, sharex='col', sharey= 'row', figsize=(10,10))
for ax, ar in zip(axs.flatten(), arrays):
    ax.imshow(ar)

但是，我得到了這個結果。

右列的第一行和第二行的寬度不相等，我希望它們相等（基本上縮小右下角的圖像，使其具有與其他圖像相同的比例）。 我已經研究了相當多的內容，但似乎沒有任何效果。 我試過tight_layout（），其他一些格式化技巧，都無濟於事...

Answer 1

您可以使用gridspec的height_ratios和width_ratios參數來設置子圖應占據的所需比例。

在這種情況下，由於對稱性，這僅僅是例如b的形狀。

import numpy as np
import matplotlib.pyplot as plt

a = np.random.rand(20, 20)
b = np.random.rand(20, 5)
c = np.random.rand(5, 20)
d = np.random.rand(5,5)
arrays = [a,b,c,d]
fig, axs = plt.subplots(2, 2, sharex='col', sharey= 'row', figsize=(10,10), 
                        gridspec_kw={"height_ratios" : b.shape, 
                                     "width_ratios" : b.shape})
for ax, ar in zip(axs.flatten(), arrays):
    ax.imshow(ar)

plt.show()

或者，更一般地說，

gridspec_kw={"height_ratios" : [a.shape[0], c.shape[0]], 
              "width_ratios" : [a.shape[1], b.shape[1]]}