Question

我试图构建一个简单的函数，它接受一个子图实例（matplotlib.axes._subplots.AxesSubplot）并将其投影转换为另一个投影，例如，转换为cartopy.crs.CRS投影之一。

这个想法看起来像这样

import cartopy.crs as ccrs
import matplotlib.pyplot as plt

def make_ax_map(ax, projection=ccrs.PlateCarree()):
    # set ax projection to the specified projection
    ...
    # other fancy formatting
    ax2.coastlines()
    ...

# Create a grid of plots
fig, (ax1, ax2) = plt.subplots(ncols=2)
# the first subplot remains unchanged
ax1.plot(np.random.rand(10))
# the second one gets another projection
make_ax_map(ax2)

当然，我可以使用fig.add_subplot()函数：

fig = plt.figure(figsize=(10,5))
ax1 = fig.add_subplot(121)
ax1.plot(np.random.rand(10))

ax2 = fig.add_subplot(122,projection=ccrs.PlateCarree())
ax2.coastlines()

但我想知道是否有一个正确的matplotlib方法来在定义之后更改子图轴投影。不幸的是，阅读matplotlib API并没有帮助。

Answer 1

您无法更改现有轴的投影，原因如下。但是，您的基本问题的解决方案只是使用matplotlib文档here中描述的subplot_kw参数plt.subplots()。例如，如果您希望所有子图都具有cartopy.crs.PlateCarree投影，则可以执行

import matplotlib.pyplot as plt
import cartopy.crs as ccrs

# Create a grid of plots
fig, (ax1, ax2) = plt.subplots(ncols=2, subplot_kw={'projection': ccrs.PlateCarree()})

关于实际问题，在创建轴集时指定投影会确定您获得的轴类，这对于每种投影类型都是不同的。例如

import matplotlib.pyplot as plt
import cartopy.crs as ccrs

ax1 = plt.subplot(311)
ax2 = plt.subplot(312, projection='polar')
ax3 = plt.subplot(313, projection=ccrs.PlateCarree())

print(type(ax1))
print(type(ax2))
print(type(ax3))

此代码将打印以下内容

<class 'matplotlib.axes._subplots.AxesSubplot'>
<class 'matplotlib.axes._subplots.PolarAxesSubplot'>
<class 'cartopy.mpl.geoaxes.GeoAxesSubplot'>

注意每个轴实际上是不同类的实例。

Answer 2

假设有多个轴用于2D绘图，例如...

fig = matplotlib.pyplot.Figure()
axs = fig.subplots(3, 4) # prepare for multiple subplots
# (some plotting here)
axs[0,0].plot([1,2,3])

...一个人可以简单地摧毁其中一个，然后换成具有3D投影的新模型：

axs[2,3].remove()
ax = fig.add_subplot(3, 4, 12, projection='3d')
ax.plot_surface(...)

请注意，与其他Python版本不同，add_subplot使用行列索引从从1 开始（而不是从0开始）。

编辑：更改了我关于索引编制的错字。

Answer 3

以下是该问题的答案：

In python, how can I inherit and override a method on a class instance, assigning this new version to the same name as the old one?

在创建斧头后，我发现了一个可以改变斧头投射的骇客，这种斧头似乎至少在下面的简单示例中有效，但是我不知道这种解决方案是否是最好的方法

from matplotlib.axes import Axes
from matplotlib.projections import register_projection

class CustomAxe(Axes):
    name = 'customaxe'

    def plotko(self, x):
        self.plot(x, 'ko')
        self.set_title('CustomAxe')

register_projection(CustomAxe)


if __name__ == '__main__':
    import matplotlib.pyplot as plt

    fig = plt.figure()

    ## use this syntax to create a customaxe directly
    # ax = fig.add_subplot(111, projection="customaxe")

    ## change the projection after creation
    ax = plt.gca()
    ax.__class__ = CustomAxe

    ax.plotko(range(10))    
    plt.show()

如何更改现有轴的matplotlib子图投影？

3 个答案: