Matplotlib:在子图的网格中重新定位子图

时间:2012-09-11 14:45:49

标签: python matplotlib customization subplot

我正在尝试制作一个包含7个子图的情节。目前我正在绘制两个列,一个有四个图,另一个有三个,例如:enter image description here

我正在按照以下方式构建这个情节:

    #! /usr/bin/env python
    import numpy as plotting
    import matplotlib
    from pylab import *
    x = np.random.rand(20)
    y = np.random.rand(20)
    fig = figure(figsize=(6.5,12))
    subplots_adjust(wspace=0.2,hspace=0.2)
    iplot = 420
    for i in range(7):
       iplot += 1
       ax = fig.add_subplot(iplot)
       ax.plot(x,y,'ko')
       ax.set_xlabel("x")
       ax.set_ylabel("y")
    savefig("subplots_example.png",bbox_inches='tight')

但是,对于发布我认为这看起来有点难看 - 我想要做的是将最后一个子图移动到两列之间的中心。那么,调整最后一个子图的位置以使其居中的最佳方法是什么?即在3X2网格中有前6个子图,下面的最后一个子图在两列之间居中。如果可能的话,我希望能够保留for循环,以便我可以简单地使用:

    if i == 6:
       # do something to reposition/centre this plot     

谢谢,

亚历

2 个答案:

答案 0 :(得分:11)

使用网格规格(doc)和4x4网格,并使每个绘图跨越2列:

import matplotlib.gridspec as gridspec
gs = gridspec.GridSpec(4, 4)
ax1 = plt.subplot(gs[0, 0:2])
ax2 = plt.subplot(gs[0,2:])
ax3 = plt.subplot(gs[1,0:2])
ax4 = plt.subplot(gs[1,2:])
ax5 = plt.subplot(gs[2,0:2])
ax6 = plt.subplot(gs[2,2:])
ax7 = plt.subplot(gs[3,1:3])
fig = gcf()
gs.tight_layout(fig)
ax_lst = [ax1,ax2,ax3,ax4,ax5,ax6,ax7]

答案 1 :(得分:8)

如果您想保留for循环,可以使用subplot2grid排列图表,这样可以生成colspan参数:

#! /usr/bin/env python
import numpy as plotting
import matplotlib
from pylab import *
x = np.random.rand(20)
y = np.random.rand(20)
fig = figure(figsize=(6.5,12))
subplots_adjust(wspace=0.2,hspace=0.2)
iplot = 420
for i in range(7):
    iplot += 1
    if i == 6:
        ax = subplot2grid((4,8), (i/2, 2), colspan=4)
    else:
        # You can be fancy and use subplot2grid for each plot, which dosen't
        # require keeping the iplot variable:
        # ax = subplot2grid((4,2), (i/2,i%2))

        # Or you can keep using add_subplot, which may be simpler:
        ax = fig.add_subplot(iplot)
    ax.plot(x,y,'ko')
    ax.set_xlabel("x")
    ax.set_ylabel("y")
savefig("subplots_example.png",bbox_inches='tight')

subplots in a grid with colspan