1

np.fabsxr.DataArray在's 但不是's上工作正常xr.Dataset。

data = xr.DataArray(np.random.randn(2, 3), coords={'x': ['a', 'b']}, dims=('x', 'y'))
ds = xr.Dataset({'foo': data, 'bar': ('x', [1, 2]), 'baz': np.pi})
np.fabs(ds)
TypeError: ufunc 'fabs' not supported for the input types, and the inputs could not be safely coerced to any supported types according to the casting rule ''safe''
np.fabs(ds['foo'])
<xarray.DataArray 'foo' (x: 2, y: 3)>
array([[ 0.384305,  0.161676,  0.07573 ],
       [ 0.789885,  1.299188,  1.965528]])
Coordinates:
  * x        (x) <U1 'a' 'b'
Dimensions without coordinates: y

任何想法如何将其应用于 a xr.Dataset？

我可以遍历xr.Dataset（见下文）中的变量，但我不确定是否有更有效的东西

for i, var in enumerate(ds.data_vars):
    ds[var] = np.fabs(ds[var])
4

2 回答 2

5

使用 do 最简单的方法是使用 Python 的内置abs()函数：abs(ds)should do it。

于 2017-10-27T18:25:01.523 回答
1

您正在寻找apply.Dataset

In [10]: ds.apply(np.fabs)
Out[10]:
<xarray.Dataset>
Dimensions:  (x: 2, y: 3)
Coordinates:
  * x        (x) <U1 'a' 'b'
Dimensions without coordinates: y
Data variables:
    foo      (x, y) float64 0.2069 2.685 1.815 1.674 1.038 0.5664
    baz      float64 3.142
    bar      (x) float64 1.0 2.0
于 2017-10-27T18:04:03.780 回答