【xarray】複数の次元を1次元に結合（平坦化）する (stack)

この記事では、以下のライブラリを用いる。

import xarray as xr

DataArray.stack()

DataArrayのメソッドstack()を使うと、複数の次元を1次元に結合することができる（平坦化）。

次のDataArrayを例として説明する。

values = [['a','b','c'],['d','e','f']]
dims = ['x','y']
coords = {'x': [0,1], 'y': [0,1,2]}
array = xr.DataArray(values, dims=dims, coords=coords)

array

使い方①

stack()の引数に、結合後の名前と結合したい次元を指定する。

stacked_array = array.stack(position=['x','y'])
stacked_array

使い方②

DataArrayのメソッドunstack()を使うと、stack()で結合した次元をもとに戻すことができる。

unstacked_array = stacked_array.unstack('position')
unstacked_array

公式ドキュメント