43

我对 Python 相当陌生,并且从更多的 Matlab 角度来看。我正在尝试制作一系列 2 x 5 面板轮廓子图。到目前为止,我的方法是(在一定程度上)将我的 Matlab 代码转换为 Python,并在循环中绘制我的子图。代码的相关部分如下所示:

fig=plt.figure(figsize=(15, 6),facecolor='w', edgecolor='k')
for i in range(10):

    #this part is just arranging the data for contourf 
    ind2 = py.find(zz==i+1)
    sfr_mass_mat = np.reshape(sfr_mass[ind2],(pixmax_x,pixmax_y))
    sfr_mass_sub = sfr_mass[ind2]
    zi = griddata(massloclist, sfrloclist, sfr_mass_sub,xi,yi,interp='nn')


    temp = 250+i  # this is to index the position of the subplot
    ax=plt.subplot(temp)
    ax.contourf(xi,yi,zi,5,cmap=plt.cm.Oranges)
    plt.subplots_adjust(hspace = .5,wspace=.001)

    #just annotating where each contour plot is being placed
    ax.set_title(str(temp))

作为这个论坛的新手,我似乎不允许附加生成的图像。但是,通过我在代码中的索引为“temp”,2 x 5 面板的结果布局为:

251 - 252 - 253 - 254 - 255
256 - 257 - 258 - 259 - 250

但是,我想要的是

250 - 251 - 252 - 253 - 254
255 - 256 - 257 - 258 - 259 

也就是说,第一个面板 (250) 出现在我认为 259 应该在的最后一个位置。251 似乎是我想要放置 250 的地方。它们似乎都处于正确的顺序中,只是循环移动了一个。

我知道这将是非常愚蠢的事情,但感谢您提供的任何帮助。

先感谢您。

4

3 回答 3

117

将您的代码与一些随机数据一起使用,这将起作用:

fig, axs = plt.subplots(2,5, figsize=(15, 6), facecolor='w', edgecolor='k')
fig.subplots_adjust(hspace = .5, wspace=.001)

axs = axs.ravel()

for i in range(10):

    axs[i].contourf(np.random.rand(10,10),5,cmap=plt.cm.Oranges)
    axs[i].set_title(str(250+i))

布局当然有点混乱,但这是因为您当前的设置(figsize、wspace 等)。

在此处输入图像描述

于 2013-06-20T10:34:34.007 回答
7

基本上与Rutger Kassies提供的解决方案相同,但使用了更 Pythonic 的语法:

fig, axs = plt.subplots(2,5, figsize=(15, 6), facecolor='w', edgecolor='k')
fig.subplots_adjust(hspace = .5, wspace=.001)

data = np.arange(250, 260)

for ax, d in zip(axs.ravel(), data):
    ax.contourf(np.random.rand(10,10), 5, cmap=plt.cm.Oranges)
    ax.set_title(str(d))
于 2016-06-19T13:44:04.847 回答
6

问题是索引subplot正在使用。子图从 1 开始计数!因此,您的代码需要阅读

fig=plt.figure(figsize=(15, 6),facecolor='w', edgecolor='k')
for i in range(10):

    #this part is just arranging the data for contourf 
    ind2 = py.find(zz==i+1)
    sfr_mass_mat = np.reshape(sfr_mass[ind2],(pixmax_x,pixmax_y))
    sfr_mass_sub = sfr_mass[ind2]
    zi = griddata(massloclist, sfrloclist, sfr_mass_sub,xi,yi,interp='nn')


    temp = 251+i  # this is to index the position of the subplot
    ax=plt.subplot(temp)
    ax.contourf(xi,yi,zi,5,cmap=plt.cm.Oranges)
    plt.subplots_adjust(hspace = .5,wspace=.001)

    #just annotating where each contour plot is being placed
    ax.set_title(str(temp))

注意计算所在行的变化temp

于 2013-06-20T10:37:21.030 回答