在Python中创建子图时调用其他函数

时间:2018-10-05 14:53:13

标签: python loops matplotlib plot subplot

更新:我现在将子图放置在正确的位置并显示正确的数据!但是,还有一个问题,那就是我不能给子图自己的标题和颜色条。实际上,我宁愿在所有图的右侧仅显示一个颜色条,但这似乎也不起作用。这是我的新代码以及一些示例数据(非常不直观,但仅用于测试目的):

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors as colors
import cartopy.crs as ccrs

lons = np.arange(-180,180,1)
lats = np.arange(-80,80,1)
dens = np.zeros((12, len(lons), len(lats)))
for i in range(12):
   dens[i,:,:] = i

def plotTitle(yr):
   letter = chr(yr-2002+97)
   return '(' + letter + ') ' + str(yr)

def DensityPlot(yr, lon, lat, dens, ax):
   Lat, Lon = np.meshgrid(lat, lon)
   density = ax.pcolormesh(Lon, Lat, dens, cmap = 'jet')
   #cbar = ax.colorbar(density, orientation='vertical', shrink=0.5, extend='both') # this gives an error: ''GeoAxesSubplot' object has no attribute 'colorbar''
   ax.coastlines()
   plt.title(plotTitle(yr),fontsize=14,fontweight='bold') # this only works for the last plot. ax.title doesn't work, gives the error ''Text' object is not callable'

fig, axes = plt.subplots(nrows=6, ncols=2, figsize=(35,35), subplot_kw={'projection': ccrs.PlateCarree()}) ## check figsize
i=0
for ax in axes.flat:
   densdata = dens[i,:,:]
   density = DensityPlot(i+2002, lons, lats, densdata, ax)
   i=i+1
   #ax.title(plotTitle(i+2002)) # this gives the same error as mentioned before: ''Text' object is not callable'
#cbar = plt.colorbar(density, orientation='vertical', shrink=0.5, extend='both')
# this gives the following RuntimeError: No mappable was found to use for colorbar creation. First define a mappable such as an image (with imshow) or a contour set (with contourf).
fig.subplots_adjust(right=0.5)

如何确保每个子图都有自己的标题,并且所有子图旁边都有一个色条?


我在用Python创建一组子图时遇到问题。我定义了一个生成特定图的函数:

def DensityPlot(lon, lat, dens):
   Lat, Lon = np.meshgrid(lat, lon)
   fig = plt.figure(figsize=(12,12))
   ax = plt.axes(projection=ccrs.PlateCarree())
   density = plt.pcolormesh(Lon, Lat, dens, cmap = 'jet')
   cbar = plt.colorbar(orientation='vertical', shrink=0.5, extend='both')
   cbar.set_label("Plastic Density ($10^{-3}$ particles km$^{-2}$)", rotation=90,fontsize=12)
   ax.coastlines()
   return density

现在,我想在for循环中使用此功能来生成子图。为简单起见,假设我可以在每个子图中为latlondens使用相同的值(因此我会多次获得相同的图)。然后,以下代码将不起作用:

fig, axes = plt.subplots(nrows=6, ncols=2)
for j in range(0,12): 
   plt.subplot(6,2,j+1)
   density = DensityPlot(lon, lat, dens)

发生以下情况:首先,显示12个空图,尽管它们的顺序正确(6x2)。此后,彼此下方出现11个空图,最后实际上以应该显示的方式显示了最后一个图。我已经检查了函数DensityPlot的代码以仅创建单个图,并且知道它可以正常工作,因此问题一定出在子图创建中。这里可能是什么问题,我该如何解决?

1 个答案:

答案 0 :(得分:1)

您自己怀疑,您是在循环外创建子图,但是每次调用函数时都会创建一个新图形。

您应该首先使用所需的投影创建图形和轴:

fig, axes = plt.subplots(nrows=6, ncols=2, subplot_kw={'projection': ccrs.PlateCarree()})

然后,您可以遍历轴,并将其作为参数传递给函数(确保删除任何图形或图形本身中的坐标轴创建):

def DensityPlot(lon, lat, dens, ax):
   Lat, Lon = np.meshgrid(lat, lon)
   density = ax.pcolormesh(Lon, Lat, dens, cmap = 'jet')
   cbar = ax.colorbar(orientation='vertical', shrink=0.5, extend='both')
   cbar.set_label("Plastic Density ($10^{-3}$ particles km$^{-2}$)", rotation=90,fontsize=12)
   ax.coastlines()

   return density

for ax in axes.flat:
    density = DensityPlot(lon, lat, dens, ax)

要在每个子图中添加标题,您需要在函数中使用ax.set_title()

ax.set_title(plotTitle(yr))

关于为多个子图添加颜色条还有很多其他问题。例如Matplotlib 2 Subplots, 1 Colorbar