如何在DataArray（xarray）中组合两个坐标值？

Question

我有一个xarray.Dataarray有4个坐标： fp, station, run_date, elnu

尺寸当前（顺序相同）：（1，2，3261，417）

站的值有 “101470” 和 “108700”，希望为了将这两个元素的尺寸合起来（1,1,3261*2,417），我想重塑它们。问题是，我无法弄清楚如何做到这一点，以及如何解决问题，然后他们会有相同的坐标（如果我会像他们将有同一站）。

也许我只是没有找到合适的词搜索谷歌（不英语）。感谢任何帮助。

Answer 1

您正在寻找的功能在xarray的.stack()方法中。例如：

>>> import xarray 

>>> import numpy as np 

>>> da = xarray.DataArray(np.zeros((1, 2, 3261, 417)), dims=['fp', 'station', 'run_date', 'elnu']) 

>>> da 
<xarray.DataArray (fp: 1, station: 2, run_date: 3261, elnu: 417)> 
array([[[[ 0., ..., 0.], 
     ..., 
     [ 0., ..., 0.]], 

     [[ 0., ..., 0.], 
     ..., 
     [ 0., ..., 0.]]]]) 
Dimensions without coordinates: fp, station, run_date, elnu 

>>> da.stack(station_date=['station', 'run_date']) 
<xarray.DataArray (fp: 1, elnu: 417, station_date: 6522)> 
array([[[ 0., 0., ..., 0., 0.], 
     [ 0., 0., ..., 0., 0.], 
     ..., 
     [ 0., 0., ..., 0., 0.], 
     [ 0., 0., ..., 0., 0.]]]) 
Coordinates: 
    * station_date (station_date) MultiIndex 
    - station  (station_date) int64 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ... 
    - run_date  (station_date) int64 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 ... 
Dimensions without coordinates: fp, elnu 

跟进.transpose()和/或.expand_dims()重新排列维度以希望的顺序。