您正在尋找的功能在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()
重新排列維度以希望的順序。