python - 计算变量的总深度

Question

我计算了潮湿的布伦特-维萨拉频率。假设变量是moistb并且维度为[height, lat, lon]。

我想绘制总深度的水平分布moistb。如何计算总深度？这个想法是对moistb每个网格点的所有深度求和。有没有办法用metpy做到这一点？

作为参考，这里有一个例子，如Schumacher 和 Johnson (2008)所示 ，他们绘制了总深度 (m) 的水平分布。

标签： pythonnumpynetcdfpython-xarraymetpy

Answer 1

在这种情况下，听起来您正在处理存储在 Xarray 中的数据DataArray。如果是这样，做你正在寻找的方法是：

moistb.sum(dim='height')

您也可以DataArray通过使用参数来使用常规 numpy 数组（或 a ）来执行此操作，该axis参数对应于按顺序排列的维数。因此，对于上面列出的顺序，这将是：

moistb.sum(axis=0)

有关更多信息，请参阅Xarray 文档或Numpy 文档。