绘制多个matplotlib轴类对象

Question

我的问题如下：我试图以可读的方式绘制6个不同的设计矩阵。 为此设计矩阵创建显示的函数是nipy模块的一部分，如下所示：

class nipy.modalities.fmri.design_matrix.DesignMatrix

  

功能show（）：设计矩阵的可视化

     

参数：

 rescale: bool, optional, 
      rescale columns magnitude for visualization or not.
 ax: axis handle, optional
      Handle to axis onto which we will draw design matrix.
 cmap: colormap, optional
      Matplotlib colormap to use, passed to imshow.

     

返回：

 ax: axis handle

基本上，我试图用3行2列和6个不同矩阵做一个子图。

n_scans = 84
tr = 7
hrf_models = ['canonical', 'canonical with derivative', 'fir', 'spm', 'spm_time', 'spm_time_dispersion']
drift_model = 'cosine'
frametimes = np.arange(0, n_scans * tr,tr)
hfcut = 128

fig1 = plt.figure()

ax1 = fig1.add_subplot(3, 2, 1)
hrf_model = hrf_models[0]
design_matrix = make_dmtx(frametimes, paradigm, hrf_model=hrf_model, drift_model=drift_model, hfcut=hfcut)
ax1 = design_matrix.show()
ax1.set_position([.05, .25, .9, .65])
ax1.set_title('Design matrix with {} as hrf_model'.format(hrf_model))

ax2 = fig1.add_subplot(3, 2, 2)
hrf_model = hrf_models[1]
design_matrix = make_dmtx(frametimes, paradigm, hrf_model=hrf_model, drift_model=drift_model, hfcut=hfcut)
ax2 = design_matrix.show()
ax2.set_position([.05, .25, .9, .65])
ax2.set_title('Design matrix with {} as hrf_model'.format(hrf_model))

......

ax6 = fig1.add_subplot(3, 2, 6)
hrf_model = hrf_models[5]
design_matrix = make_dmtx(frametimes, paradigm, hrf_model=hrf_model, drift_model=drift_model, hfcut=hfcut)
ax6 = design_matrix.show()
ax6.set_position([.05, .25, .9, .65])
ax6.set_title('Design matrix with {} as hrf_model'.format(hrf_model))

plt.show()

目前输出的是3行2列的数字，上面有空白图表，然后每个设计矩阵单独显示。

此外，列表hrf_models上的循环比重复相同块的6倍要好得多。我在某个时候做到了，但输出结果完全一样悲伤。

当前输出（需要滚动才能看到所有设计矩阵）：

感谢您的帮助！

Answer 1

基本上，您在问题中添加的文档字符串的摘录已经告诉您解决方案。您需要将ax参数用于DesignMatrix.show()

ax1 = fig1.add_subplot(3, 2, 1)
design_matrix = make_dmtx(...)
design_matrix.show(ax = ax1)

要使用循环，您可以先生成所有轴，然后循环它们。

fig, axes = plt.subplots(nrows=3,ncols=2)
for i, ax in enumerate(axes.flatten()):
    hrf_model = hrf_models[0]
    design_matrix = make_dmtx(frametimes, paradigm, hrf_model=hrf_models[i], 
                              drift_model=drift_model, hfcut=hfcut)
    design_matrix.show(ax = ax)

请注意，我还没有在这里测试任何东西，因为我没有nipy可用。