时间戳

首页 > 解决方案 > xarray- 向 DataArray 添加一个新维度并将坐标分配给该维度

numpy - xarray- 向 DataArray 添加一个新维度并将坐标分配给该维度

问题描述

我有一个名为的 DataArray 对象test。它包含一个名为的变量FFDI 90TH PERCENTILE以及纬度和经度维度。

import numpy as np
import pandas as pd
import xarray as xr

print(test)

<xarray.DataArray 'FFDI 90TH PERCENTILE' (latitude: 106, longitude: 193)>
array([[ 2.699949,  2.699277,  2.677113, ...,  3.353225,  3.381503,  3.392549],
       [ 2.7     ,  2.704608,  2.70228 , ...,  3.422083,  3.435692,  3.465664],
       [ 2.720069,  2.71194 ,  2.711843, ...,  3.5     ,  3.5     ,  3.501185],
       ...,
       [34.863322, 34.825574, 34.694171, ...,  8.599811,  8.50329 ,  8.815733],
       [34.728609, 35.180146, 35.203714, ...,  8.164053,  8.01015 ,  7.94335 ],
       [34.654186, 34.865241, 34.987067, ...,  7.814975,  7.644326,  7.925   ]])
Coordinates:
  * latitude   (latitude) float32 -39.2 -39.149525 ... -33.950478 -33.9
  * longitude  (longitude) float32 140.8 140.84792 140.89584 ... 149.95209 150.0

我有以下times元素：

times = pd.date_range("1972/12/01","2017/12/01",freq='D',closed='left')
time_da = xr.DataArray(times, [('time', times)])

我想添加一个新维度并调用它time；并将上述时间分配给time维度作为坐标。所以新的testDataArray 看起来像：

<xarray.DataArray 'FFDI 90TH PERCENTILE' (time: 16436, latitude: 106, longitude: 193)>

我用assign_coords和做了以下尝试expand_dims。他们都没有工作。

一：

test_assigned = test.assign_coords({'time': times.values})
TypeError: assign_coords() takes 1 positional argument but 2 were given

二：

test_assigned = test.assign_coords(time=times.values)
ValueError: cannot add coordinates with new dimensions to a DataArray

标签： numpypython-xarray

解决方案

这是 的用例expand_dims，它将沿新维度扩展数组并为其分配坐标（如果提供）：

result = test.expand_dims(time=time_da)

推荐阅读