Python中如何绘制连接相对涡度(relative vorticity)空间图各经度最大值点的线？

首先明确：绘制这条最大值连线不需要二维数组，只需要每个经度对应的最大值纬度（一维数组）和经度数组（一维）即可，matplotlib的plot函数可以直接接收两个一维数组作为x、y坐标生成空间线。

假设你的数据结构是：

• lat: 一维数组，存储所有纬度值
• lon: 一维数组，存储所有经度值
• vorticity: 二维数组，形状为(纬度数, 经度数)，对应每个经纬度点的相对涡度值

步骤1：提取每个经度的最大值对应纬度 #

用numpy的argmax函数，沿纬度轴（axis=0）找到每个经度列中涡度最大的索引，再映射到实际纬度值：

import numpy as np
import matplotlib.pyplot as plt

# 替换成你的真实数据
lat = np.linspace(-90, 90, 180)
lon = np.linspace(0, 360, 360)
vorticity = np.random.randn(len(lat), len(lon))  # 模拟涡度数据

# 对每个经度（列），找到涡度最大值对应的纬度索引
max_lat_idx = np.argmax(vorticity, axis=0)
# 获取每个经度对应的最大值纬度
max_lat_values = lat[max_lat_idx]

步骤2：绘制涡度图与最大值连线 #

先绘制原始的涡度填充图，再用plot函数叠加连线：

# 绘制涡度填充图
plt.contourf(lon, lat, vorticity, cmap='RdBu_r', levels=20)
plt.colorbar(label='Relative Vorticity')

# 绘制最大值连线
plt.plot(lon, max_lat_values, color='black', linewidth=2, label='Max Vorticity Line')
plt.scatter(lon, max_lat_values, color='white', s=15, zorder=5)  # 可选：标记每个最大值点

# 设置图属性
plt.xlabel('Longitude')
plt.ylabel('Latitude')
plt.title('Relative Vorticity with Longitude-wise Max Line')
plt.legend()
plt.show()

注意事项 #

• 如果你的涡度数组形状是(经度数, 纬度数)（轴顺序相反），需要把argmax的axis参数改成1，调整为max_lat_idx = np.argmax(vorticity, axis=1)，同时后续索引也要对应调整。
• 若同一经度存在多个涡度相等的最大值，argmax只会返回第一个出现的纬度索引，若需要处理这种情况，可以用np.where(vorticity == np.max(vorticity, axis=0))获取所有最大值点的索引。

内容的提问来源于stack exchange，提问作者Riya