以太平洋为中心的世界地图上的等高线图

2024-04-24 06:13:06 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图在世界地图上绘制一些数据,这些数据可以集中在大西洋附近(即180°W–180°e)或太平洋(即0°–360°)。以下是程序(带有虚拟数据):

import argparse

import numpy as np
import matplotlib.pyplot as plt
import cartopy.crs as ccrs
from cartopy.mpl.gridliner import LONGITUDE_FORMATTER, LATITUDE_FORMATTER

parser = argparse.ArgumentParser()
parser.add_argument('--center', choices=['atlantic', 'pacific'], default='atlantic')
parser.add_argument('--outfile', default='plot.png')
args = parser.parse_args()

lat = np.linspace(-89.95, 89.95, 1800)
if args.center == 'atlantic':
        lon = np.linspace(-179.95, 179.95, 3600)
        clon = 0
else:
        lon = np.linspace(0.05, 359.95, 3600)
        clon = 180
x, y = np.meshgrid(lon, lat)
z = np.sin(x / 180 * np.pi) * np.sin(y / 180 * np.pi)

fig = plt.figure(figsize=(21, 7))
crs = ccrs.PlateCarree(central_longitude=clon)
ax = plt.axes(projection=crs)
ax.coastlines(resolution='110m', color='white', linewidth=2)

gl = ax.gridlines(crs=crs, draw_labels=True, linewidth=1, color='black', linestyle='--')
gl.xformatter = LONGITUDE_FORMATTER
gl.yformatter = LATITUDE_FORMATTER
gl.xlabel_style = {'size': 16, 'color': 'black'}
gl.ylabel_style = {'size': 16, 'color': 'black'}

plt.contourf(x, y, z, cmap='RdYlBu_r')
cb = plt.colorbar(ax=ax, orientation='vertical', pad=0.02, aspect=16, shrink=0.8)
cb.ax.tick_params(labelsize=16)

fig.savefig(args.outfile, bbox_inches='tight', pad_inches=0.1)

但是,当我从--center=atlantic切换到--center=pacific时,只有海岸线移动,而X轴和数据不移动,导致绘图不一致。(根据我虚构的数据,北美应该是蓝色的,亚洲应该是红色的。)

  • --center=atlantic:atlantic.png
  • --center=pacificpacific.png

我怎样才能画出一个以太平洋为中心的正确的图呢


Tags: 数据importparserformatterasnpargsplt
1条回答
网友
1楼 · 发布于 2024-04-24 06:13:06

看起来我需要以下更改:

  • 拥有一个普通的PlateCarree对象(除了现有的带有central_longitude集的对象之外),并在除调用plt.axes之外的所有情况下使用它。(我不明白为什么,但我发现它是有效的。)
  • 添加对ax.set_extent的调用,也使用vanilla PlateCarree对象
  • plt.contourf中使用transform,也可以与vanilla PlateCarree对象一起使用

以下是与原始代码的区别:

@@ -23,0 +24 @@
+crs0 = ccrs.PlateCarree()
@@ -25,0 +27 @@
+ax.set_extent([lon[0], lon[-1], lat[0], lat[-1]], crs=crs0)
@@ -28 +30 @@
-gl = ax.gridlines(crs=crs, draw_labels=True, linewidth=1, color='black', linestyle=' ')
+gl = ax.gridlines(crs=crs0, draw_labels=True, linewidth=1, color='black', linestyle=' ')
@@ -34 +36 @@
-plt.contourf(x, y, z, cmap='RdYlBu_r')
+plt.contourf(x, y, z, cmap='RdYlBu_r', transform=crs0)

这会产生180°W180°E相互覆盖。作为一个快速解决方案,我这样做:

import matplotlib.ticker as mticker

# Fix LONGITUDE_FORMATTER so that either +180 or -180 returns just '180°',
# instead of '180°E' or '180°W'.
LONGITUDE_FORMATTER_NEW = mticker.FuncFormatter(
       lambda v, pos: '180\u00B0' if abs(v) == 180 else LONGITUDE_FORMATTER.func(v, pos)
)

因此,相同的字符串180°被覆盖在彼此顶部的相同位置,从而最小化问题的视觉效果

LONGITUDE_FORMATTER无法处理任何超出[−180,+180]也不错,但我选择不在这里讨论。)

结果如下:

  • center=atlantic:atlantic.png
  • center=pacificpacific.png

相关问题 更多 >