【气象】python计算平流并cartopy绘图（计算差分、弧度制转化、广播运算）

原创

已于 2023-02-23 14:57:52 修改 · 3.7k 阅读

标签

#python #numpy

收录于

于 2022-11-28 14:18:07 首次发布

本文介绍了如何使用Python和xarray库处理大气环流数据，通过T_advection类实现风场与温度场的advection计算，并展示了平流公式和关键计算步骤，包括数据读取、差分运算和地图绘制。

一、总代码：

import xarray as xr
import numpy as np

import cartopy.crs as ccrs
import cartopy.feature as cfeature
import matplotlib.pyplot as plt
import cartopy.mpl.ticker as cticker
from cartopy.util  import  add_cyclic_point

class T_advection(object):
    def __init__(self,file_u,file_v,file_t):
        self.f_u = xr.open_dataset(file_u)
        self.f_v = xr.open_dataset(file_v)
        self.f_t = xr.open_dataset(file_t)

        pass

    def cal_advection(self):
        a = 6370000
        u0 = self.f_u.uwnd.loc['2014-11-01',850,90:0,0:180]
        v0 = self.f_v.vwnd.loc['2014-11-01',850,90:0,0:180]
        t = self.f_t.air.loc['2014-11-01',850,90:0,0:180]
        lon0 = u0.lon
        lat0 = u0.lat
        dlon = (np.gradient(lon0)*np.pi/180).reshape((1,-1))
        dlat = (np.gradient(lat0)*np.pi/180).reshape((-1,1))
        coslat = (np.array(np.cos((lat0)*np.pi/180))).reshape((-1,1))
        dx = a*coslat*dlon
        dy = a*dlat