python - metpy.units 函数有问题吗？

Question

我正在尝试使用 metpy.calc 函数，但是每次我尝试使用 metpy.units 分配单位时，它都不会运行。如果我允许的话，它会坐在那里并继续运行几个小时。过去，metpy 函数在相同的数据集上对我来说工作得非常好，但由于某种原因将不再工作。我尝试过更新metpy和spyder，并且还使用了多台笔记本电脑以确保它不仅仅发生在一台笔记本电脑上。我还尝试了两种不同的方法 metpy.units 。它使用单位(“kg/kg”)运行，但实际上不会将单位分配给数组。关于如何让它发挥作用有什么想法吗？

这是我的代码:

import numpy as np 
import matplotlib.pyplot as plt
from netCDF4 import Dataset
import metpy
import metpy.calc as mpcalc
from metpy.units import units

f = Dataset('C:/Users/hkenward/Desktop/MS/cm1out.nc', mode = 'r', format="NETCDF4")
mrv=f.variables['qv'][:]
pt=f.variables['th'][:]
p=f.variables['prs'][:]
p=p/100.
mrv=mrv*units('kg/kg')
p=p * units.hPa
pt=pt*units.kelvin

e=mpcalc.vapor_pressure(p,mrv)
Td=mpcalc.dewpoint(e)
T=mpcalc.temperature_from_potential_temperature(p, pt)
Te=mpcalc.equivalent_potential_temperature(p, T, Td)

Answer 1

这是 Pint 单位框架(由 MetPy 使用)和屏蔽数组(netCDF4-python 默认情况下吐出的内容)的兼容性问题。最简单的解决方法是将左侧的单位相乘，如下所示:

pt = units.kelvin * pt

这使得一切正常。希望通过使用 Xarray 之类的东西，这样的事情将来会变得更好。