Обрезать pcolormesh за пределами круговой set_boundary в Cartopy - PullRequest
0 голосов
/ 18 июня 2019

Я использую Cartopy для своих полярных исследований и хотел бы обвести круговую границу вокруг моих данных, которые я нанесу в проекции NorthPolarStereo(). Я использую set_extent, чтобы указать, с какой широты я хотел бы построить свои данные, и использую set_boundary для создания круговой границы, как объяснено в галерее . Затем я использую matplotlib.pyplot.pcolormesh для построения фактических данных. Однако, скажем, я использую set_extent, чтобы определить минимальную широту 55 градусов, некоторые из моих данных ниже 55 градусов все еще выводятся за пределы моего set_boundary. Как мне обрезать эти данные?

map_crs = ccrs.NorthPolarStereo(central_longitude=0.0, globe=None)

# Build axes
fig     = plt.figure()
ax      = plt.axes(projection=map_crs)

plotfield = ax.pcolormesh(lons, lats, data, transform=ccrs.PlateCarree())
ax.set_extent((-180, 180, 55, 90), crs=ccrs.PlateCarree())
gl = ax.gridlines()

# Circular clipping
theta = np.linspace(0, 2*np.pi, 400)
center, radius = [0.5, 0.5], 0.5
verts = np.vstack([np.sin(theta), np.cos(theta)]).T
circle = mpath.Path(verts * radius + center)
ax.set_boundary(circle, transform=ax.transAxes)

data outside of set_boundary

1 Ответ

1 голос
/ 18 июня 2019

У меня нет картопии, чтобы проверить его в тех же условиях, что и у вас, но вы можете обрезать pcolormesh, используя объект Patch любой формы:

# the code below is adapted from the pcolormesh example
# https://matplotlib.org/3.1.0/gallery/images_contours_and_fields/pcolormesh_levels.html#sphx-glr-gallery-images-contours-and-fields-pcolormesh-levels-py

# make these smaller to increase the resolution
dx, dy = 0.05, 0.05

# generate 2 2d grids for the x & y bounds
y, x = np.mgrid[slice(1, 5 + dy, dy),
                slice(1, 5 + dx, dx)]
z = np.sin(x)**10 + np.cos(10 + y*x) * np.cos(x)

theta = np.linspace(0, 2*np.pi, 400)
center, radius = [0.5, 0.5], 0.5
verts = np.vstack([np.sin(theta), np.cos(theta)]).T
circle = matplotlib.path.Path(verts * radius + center)

fig, ax = plt.subplots()
im = ax.pcolormesh(x, y, z, cmap='viridis', clip_path=(circle, ax.transAxes))

fig.colorbar(im, ax=ax)
fig.tight_layout()

plt.show()

enter image description here

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...