python - 次元を展開 xarray

Question

オブジェクトの次元 (および座標) を拡張する既存の方法またはアプローチはありxarray.DataArrayますか?

np.expand_dimsに似たものを取得したいと同時に、新しい Expanded の新しい次元と座標変数を定義しDataArrayます。

I を使用DataArray.assign_coords()すると、新しい座標変数を作成できますが、配列自体は新しい軸で展開されません。

Answer 1

これを行うための何らかの方法が役立つことに同意します。現在、xarray には直接存在しませんが、GitHubに問題を提出して、新しい機能の API について議論したり、それを実装するプル リクエストを作成したりすることをお勧めします。

新しいxarray.broadcast関数には、この目的に十分ないくつかの関連機能が含まれています。

import xarray as xr
import numpy as np
data = xr.DataArray([1, 2, 3], dims='x')
other = xr.DataArray(np.zeros(4), coords=[('y', list('abcd'))])
data2, other2 = xr.broadcast(data, other)
print(data2)
# <xarray.DataArray (x: 3, y: 4)>
# array([[1, 1, 1, 1],
#       [2, 2, 2, 2],
#       [3, 3, 3, 3]])
# Coordinates:
#   * x        (x) int64 0 1 2
#   * y        (y) |S1 'a' 'b' 'c' 'd'