我试图创建一个由2x2网格组成的图形,其中每个象限中有2个垂直堆叠的子图(即2x1网格)。但我似乎无法弄清楚如何实现这一目标。
我最接近的是使用gridspec和一些丑陋的代码(见下文),但由于gridspec.update(hspace=X)
改变了所有子图的间距,我仍然不在我所在的位置。我喜欢。
理想情况下,我想要的是,使用下面的图片作为示例,减少每个象限内的子图之间的间距,同时增加顶部和底部象限之间的垂直间距(即1-3到2-4之间) 。
有没有办法做到这一点(使用或不使用gridspec)?我最初想象的是生成每个子子图网格(即每个2x1网格)并将它们插入到更大的2x2子图中网格中,但我还没有弄清楚如何将子图添加到子图中,如果有的话甚至是一种方式。
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec
plt.figure(figsize=(10, 8))
gs = gridspec.GridSpec(4,2)
gs.update(hspace=0.4)
for i in range(2):
for j in range(4):
ax = plt.subplot(gs[j,i])
ax.spines['top'].set_visible(False)
ax.spines['right'].set_visible(False)
plt.tick_params(which='both', top='off', right='off')
if j % 2 == 0:
ax.set_title(str(i+j+1))
ax.plot([1,2,3], [1,2,3])
ax.spines['bottom'].set_visible(False)
ax.get_xaxis().set_visible(False)
else:
ax.plot([1,2,3], [3,2,1])
答案 0 :(得分:36)
你可以nest your GridSpec using SubplotSpec。外部网格为2 x 2,内部网格为2 x 1.以下代码应为您提供基本概念。
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec
fig = plt.figure(figsize=(10, 8))
outer = gridspec.GridSpec(2, 2, wspace=0.2, hspace=0.2)
for i in range(4):
inner = gridspec.GridSpecFromSubplotSpec(2, 1,
subplot_spec=outer[i], wspace=0.1, hspace=0.1)
for j in range(2):
ax = plt.Subplot(fig, inner[j])
t = ax.text(0.5,0.5, 'outer=%d, inner=%d' % (i,j))
t.set_ha('center')
ax.set_xticks([])
ax.set_yticks([])
fig.add_subplot(ax)
fig.show()
答案 1 :(得分:4)
您现在可以使用 figure.subfigure
嵌套子图:https://matplotlib.org/stable/gallery/subplots_axes_and_figures/subfigures.html
您也可以很好地使用 subplot_mosaic,因为它支持嵌套: https://matplotlib.org/stable/tutorials/provisional/mosaic.html
虽然嵌套总是有点痛苦,但可以说这两种方法都使过程更容易、更灵活。
取自上述示例:
import matplotlib.pyplot as plt
fig = plt.figure(constrained_layout=True)
subfigs = fig.subfigures(2, 2)
for outerind, subfig in enumerate(subfigs.flat):
subfig.suptitle(f'Subfig {outerind}')
axs = subfig.subplots(2, 1)
for innerind, ax in enumerate(axs.flat):
ax.set_title(f'outer={outerind}, inner={innerind}', fontsize='small')
ax.set_xticks([])
ax.set_yticks([])
plt.show()
答案 2 :(得分:0)
由于我不得不依靠Suever的回答,但必须根据自己的需要进行修改,所以我认为如果有人觉得有帮助,我可能会做出贡献:
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec
channelArrangement = [16, 17, 18 , 19 , 22, 25, 28 , 29 , 31]
fig = plt.figure(figsize=(10, 8))
outer = gridspec.GridSpec(1, 2, wspace=0.2, hspace=0.2)
for i in range(2):
inner = gridspec.GridSpecFromSubplotSpec(3, 3, subplot_spec=outer[i], wspace=0.1, hspace=0.1)
row = 0
col = 0
maxCol = 3
for chan in channelArangement:
ax = plt.Subplot(fig, inner[row,col])
t= ax.text(0.5,0.5, 'outer=%d\nrow=%d\ncol=%d' % (i,row,col))
ax.set_xticks([])
ax.set_yticks([])
t.set_ha('center')
fig.add_subplot(ax)
col += 1
if col == maxCol:
col = 0
row += 1
plt.show()