for循环内的子图

时间:2019-04-13 13:38:59

标签: python loops for-loop subplot

我想从一个循环中将四个文件图放入一个子图图中。

我已经循环播放了一些文件(四个数据文件)以从中提取某些数据(例如,纬度,经度和气溶胶光学深度),并正确打印了该文件,这表明它已成功循环浏览了文件并提取了我需要的数据。当我继续循环并仅绘制结果时,它会绘制四个单独的图形,因为我尚未完成子图的绘制。

当我实现一些子图代码时,它会做很多事情,但不是我想要的。我想要1个具有2行2列的子图来显示每个文件,而不是4个子图只为每个子图重复一个文件。

这是我的压缩代码:

for lim in mlims:
   fil= ilfil + innm +'kd00' + jad +'nam'
   ln= Dataset(ilfil)
   longit = vn.variables['longitude'][:]
   latitud = vn.variables['latitude'][:] 
   ## etc .....
   aode       = var1+var2
   aod=np.squeeze(aode[:,2,:,:])

   lons, lats = np.meshgrid(lonitud, latatit)
   x, y = map(lons, lats)

   ii=[0,1,2,3]
   fig, axes = plt.subplots(nrows=4, ncols=4, figsize=(2,5))
   for ax,mon_index,lname in zip(axes.flatten(),ii, mnames):


       axis=np.arange(0+0.025,0.5+0.025,0.025)

       cs = ax.contourf(x,y,aod,axis,cmap='seismic',linewidths=1.)
       cbar = map.colorbar(cs)

       plt.title(AOD)

plt.show()

代码输出以下内容:1是文件的图,但是我不希望同一文件在一个子图图中重复,而2+是图。

我希望它输出:

1 1 1 2-第一个文件

1 1 1 3-第二个文件

1 1 1 4-第三

1 1 1 5

我想要得到的只是: 2 3 4 5

1 个答案:

答案 0 :(得分:0)

制作图形和子图;遍历文件和轴;在 current 轴上绘制文件中的数据。这是一个显示过程的玩具示例。

files = [range(3),range(4),range(5),range(6)]
fig, axarr = plt.subplots(2,2)
for data, ax in zip(files, axarr.flat):
    # if you are looping over filenames
    # process/extract the data from the file here
    ax.plot(data)

subplots demo from Matplotlib docs

.subplots()返回一个图形和一个子图Axes的数组。

>>> axarr
array([[<matplotlib.axes._subplots.AxesSubplot object at 0x000000000D2BA358>,
        <matplotlib.axes._subplots.AxesSubplot object at 0x000000000F3F29B0>],
       [<matplotlib.axes._subplots.AxesSubplot object at 0x000000000F429080>,
        <matplotlib.axes._subplots.AxesSubplot object at 0x000000000F44F710>]],
      dtype=object)

您可以通过将索引到(?)数组中来选择它们,而不是依次遍历这些轴。

>>> axarr[0,1]

<matplotlib.axes._subplots.AxesSubplot object at 0x000000000F3F29B0>
>>>

pyplot具有当前轴的概念。调用.subplot()方法将使轴成为当前,然后后续语句将对该轴起作用。这是乱序操作单个子图的另一种方法。

plt.subplot(rows,cols,1)
plt.plot(range(10),'r')
plt.subplot(rows,cols,3)
plt.plot(range(100),'b')
plt.subplot(rows,cols,2)
plt.plot(range(5),'o')
plt.subplot(rows,cols,4)
plt.plot(range(5),'g')