使用底图会破坏 gridspec 索引

Using basemap breaks gridspec indexing

我正在尝试使用 gridspec 绘制两组地图,当我不包括底图时格式完美,但一旦我这样做,第一张图像在其他图后面显示得更大,并且网格以第二张图片并依次显示,最后一张图为空。

绘制正确(https://imgur.com/a/YNVOIgc):

包含底图,绘制不正确 (https://imgur.com/a/uYfohiJ)

这是我重现问题的代码,要查看它是否正常工作,只需注释掉 m.drawcoastlines() 行。

import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec
from mpl_toolkits.basemap import Basemap

outer_grid = gridspec.GridSpec(2, 1, wspace=0.2, hspace=0.2)

#create plot
fig = plt.figure(figsize=(10, 8))
i=0

while i<2:
    inner_grid = gridspec.GridSpecFromSubplotSpec(3, 3, subplot_spec=outer_grid[i], wspace=0.0, hspace=0.0) 
    n=0
    while n<9:
        ax = plt.Subplot(fig, inner_grid[n])
        n=n+1
        m = Basemap(lat_0=0,lon_0=0)
        m.drawcoastlines()
        fig.add_subplot(ax)
    i=i+1

plt.suptitle('Gridspec/Basemap') 
plt.show()

你不应该使用 plt.Subplot。而是在添加时直接创建子图

ax = fig.add_subplot(inner_grid[n])

虽然不是绝对必要,但将轴作为参数添加到 Basemap 可能会有所帮助;这可能有助于记住实际使用了哪些轴。

m = Basemap(lat_0=0,lon_0=0, ax=ax)

完整代码:

import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec
from mpl_toolkits.basemap import Basemap

outer_grid = gridspec.GridSpec(2, 1, wspace=0.2, hspace=0.2)

#create plot
fig = plt.figure(figsize=(10, 8))
i=0

while i<2:
    inner_grid = gridspec.GridSpecFromSubplotSpec(3, 3, subplot_spec=outer_grid[i], wspace=0.0, hspace=0.0) 
    n=0
    while n<9:
        ax = fig.add_subplot(inner_grid[n])
        n=n+1
        m = Basemap(lat_0=0,lon_0=0, ax=ax)
        m.drawcoastlines()

    i=i+1

plt.suptitle('Gridspec/Basemap') 
plt.show()