应用numpy.polyfit到xarray数据

Question

Xarray是否支持numpy计算函数，比如polyfit？或者有没有一种有效的方法将这些函数应用于数据集？在

例子：我想计算一条直线的斜率，它与两个变量（温度和高度）相匹配，以计算递减率。我有一个数据集（下图），包含这两个维度为（垂直、时间、xgrid_0、ygrid_0）的变量。在

<xarray.dataset>
Dimensions:    (PressLev: 7, time: 48, xgrid_0: 685, ygrid_0: 485)
Coordinates:
    gridlat_0  (ygrid_0, xgrid_0) float32 44.6896 44.6956 44.7015 44.7075 ...
    gridlon_0  (ygrid_0, xgrid_0) float32 -129.906 -129.879 -129.851 ...
  * ygrid_0    (ygrid_0) int64 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 ...
  * xgrid_0    (xgrid_0) int64 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 ...
  * time       (time) datetime64[ns] 2016-08-15T01:00:00 2016-08-15T02:00:00 ...
  * PressLev   (PressLev) int64 0 1 2 3 4 5 6
Data variables:
    Temperature       (PressLev, time, ygrid_0, xgrid_0) float64 289.4 289.4 289.4 ...
    Height       (PressLev, time, ygrid_0, xgrid_0) float64 85.23 85.13 84.98 ...

如果我提取给定时间的温度和高度，xgrid，ygrid，0；我可以使用纽比.波利菲特功能。在

^{pr2}$

但这是一种缓慢而低效的方法。有什么更好的方法来解决这个问题吗？在