我正在使用一个新的 Python 包 (metpy),其中包含几个子包 - 其中一个使用以下代码定义梯度和风收敛:
from __future__ import division
import numpy as np
from ..package_tools import Exporter
from ..constants import g
from ..units import atleast_2d, concatenate, units
### the 'units' and 'constants' packages are within metpy
exporter = Exporter(globals())
def _gradient(f, *args, **kwargs):
if len(args) < f.ndim:
args = list(args)
args.extend([units.Quantity(1., 'dimensionless')] * (f.ndim - len(args)))
grad = np.gradient(f, *args, **kwargs)
if f.ndim == 1:
return units.Quantity(grad, f.units / args[0].units)
return [units.Quantity(g, f.units / dx.units) for dx, g in zip(args, grad)]
def _stack(arrs):
return concatenate([a[np.newaxis] for a in arrs], axis=0)
def _get_gradients(u, v, dx, dy):
# Helper function for getting convergence and vorticity from 2D arrays
dudx, dudy = _gradient(u, dx, dy)
dvdx, dvdy = _gradient(v, dx, dy)
return dudx, dudy, dvdx, dvdy
def h_convergence(u, v, dx, dy):
dudx, _, _, dvdy = _get_gradients(u, v, dx, dy)
return dudx + dvdy
当我尝试使用函数 h_convergence 时,我在这一行收到以下错误:
if f.ndim == 1:
return units.Quantity(grad, f.units / args[0].units)
return [units.Quantity(g, f.units / dx.units) for dx, g in zip(args, grad)]
AttributeError: 'numpy.ndarray' object has no attribute 'units'
我是 Python 的新手,但据我所知,它试图从 metpy.units 包中调用“units”,而不是从 numpy?任何人都可以建议吗?这不是我的代码,但我不知道为什么当我调用 h_convergence 函数时它不起作用。