将 numpy 数组作为变量添加到 Xarray 数据集

Question

这似乎是一个非常基本的操作，但我无法弄清楚如何使用 xarray 文档进行操作。

我有一个 xarray 数据集：

dss
<xarray.DataArray (y: 1000, x: 1334)>
dask.array<shape=(1000, 1334), dtype=uint8, chunksize=(222, 58)>
Coordinates:
    band     int32 1
  * y        (y) float64 2.218e+06 2.218e+06 2.218e+06 2.218e+06 2.218e+06 ...
  * x        (x) float64 1.891e+06 1.891e+06 1.891e+06 1.891e+06 1.891e+06 ...
Attributes:
    transform:   (30.0, 0.0, -2493045.0, 0.0, -30.0, 3310005.0, 0.0, 0.0, 1.0)
    crs:         +ellps=GRS80 +lat_0=23 +lat_1=29.5 +lat_2=45.5 +lon_0=-96 +n...
    res:         (30.0, 30.0)
    is_tiled:    1
    nodatavals:  (nan,)

和一个具有正确维度的 numpy 数组：

print(np.shape(nmap))
(1000, 1334)
nmap
array([[ 0.15,  0.1 ,  0.15, ...,  0.05,  0.05,  0.02],
       [ 0.15,  0.1 ,  0.05, ...,  0.05,  0.05,  0.05],
       [ 0.1 ,  0.15,  0.15, ...,  0.05,  0.05,  0.02],
       ..., 
       [ 0.02,  0.02,  0.02, ...,  0.02,  0.02,  0.02],
       [ 0.02,  0.09,  0.09, ...,  0.02,  0.02,  0.02],
       [ 0.02,  0.09,  0.09, ...,  0.02,  0.02,  0.02]])

我想将数组添加到数据集。我的最终目标是使用x和y提取nmap新网格上的内插值进行空间插值。

Answer 1

假设您想将温度数据添加到数据集中：

# Take the following dataset as an example
data_set=xr.Dataset( coords={'lon': (['x', 'y'], lon),
                    'lat': (['x', 'y'], lat),
                    'time': pd.date_range('2014-09-06', periods=3)})
temp=np.array([[25, 24, 20, -12],[23, 21, 22, -11]])
data_set["Temperature"]=(['x', 'y', 'time'],  temp)

Answer 2

你想创建一个包含你的 numpy 数组 nmap 的数据集吗？或者你想做一个算术dss + nmap？

对于前一种情况下，你需要做一个Dataset从dss第一和分配nmap给它，因为你dss不是一个Dataset而是一个DataArray。

为了使Dataset从DataArrayS，你可以传递一个字典映射数组名到DataArray中的对象。如果您的数组不是DataArraynumpy 数组或 dask 数组，则需要一个元组（维度、数组、[属性]）。

ds = xr.Dataset({'dss': dss, 'nmap': (('y', 'x'), nmap)})

或者做同样事情的另一种方法是

ds = xr.Dataset({})
ds['dss'] = ds
ds['nmap'] = (('y', 'x'), nmap)

对于后一种情况，只需执行

dss + nmap