就地舍入和覆盖 xarray 数据

Question

我想将 xarray 中的所有值四舍五入到小数点后两位。

我尝试了以下方法：

def round_dim(data_arr):
   data_arr.data.round(decimals = 2)
   return data_arr

我这样称呼它：

data_values = <xarray.core.dataarray.DataArray>
data_values = round_dim(data_values)

我希望 data_values 现在如何将相同的值四舍五入到小数点后两位，但这并没有发生。

Answer 1

如果你想将round(decimals = 2)应用于你的整个数据集，你可以使用xarray.Dataset.map。

它返回一个复制原始数据集的所有数据都四舍五入了，它不能就地工作。

import numpy as np
import xarray

dataset = xarray.Dataset()
dataset["a"] = xarray.DataArray(np.linspace(0, 1, 10))
dataset["b"] = xarray.DataArray(np.arange(10))
print(dataset)
# <xarray.Dataset>
# ...
#    a        (dim_0) float64 0.0 0.1111 0.2222 0.3333 ... 0.7778 0.8889 1.0
#    b        (dim_0) int32 0 1 2 3 4 5 6 7 8 9

def round_dim(dataset):
    return dataset.map(lambda a: np.round(a, decimals=2))

dataset = round_dim(dataset)
print(dataset)
# <xarray.Dataset>
# ...
#     a        (dim_0) float64 0.0 0.11 0.22 0.33 0.44 0.56 0.67 0.78 0.89 1.0
#     b        (dim_0) int32 0 1 2 3 4 5 6 7 8 9