python - xarray.DataArray.expand_dims 应该做什么？

Question

我正在尝试向 DataArray 添加一个具有一定长度的新维度。

我尝试使用expand_dims方法，其文档表明我应该能够做到这一点

http://xarray.pydata.org/en/stable/generated/xarray.DataArray.expand_dims.html#xarray.DataArray.expand_dims

“新变量中包含的维度。如果作为 str 或 str 序列提供，则插入长度为 1 的维度。如果作为字典提供，则键是新维度，值是整数 (< strong>给出新维度的长度)或sequence/ndarray(给出新维度的坐标)。”

我尝试过，但没有成功。

test = xr.DataArray([1,2,3,4,5,6])

test
Out[114]: 
<xarray.DataArray (dim_0: 6)>
array([1, 2, 3, 4, 5, 6])
Dimensions without coordinates: dim_0

test.expand_dims({'dim_1':3})
Out[115]: 
<xarray.DataArray (dim_1: 1, dim_0: 6)>
array([[1, 2, 3, 4, 5, 6]])
Dimensions without coordinates: dim_1, dim_0

我看到数组中的附加维度来自额外的括号，但长度为一:

test.expand_dims({'dim_1':3}).shape
Out[120]: (1, 6)

我想我期望形状是 (3,6)(扩展数据)或 (3,2)( reshape 数据)。

我做错了什么或不理解吗？我刚刚发现了 xarray，所以这完全有可能是我没有得到的概念性的东西。

Answer 1

试试这个:

test.expand_dims(dim_1 = 3)
Out: 
<xarray.DataArray (dim_1: 3, dim_0: 6)>
array([[1, 2, 3, 4, 5, 6],
       [1, 2, 3, 4, 5, 6],
       [1, 2, 3, 4, 5, 6]])
Dimensions without coordinates: dim_1, dim_0

test.expand_dims(dim_1 = 3).shape
Out: (3, 6)

我提到了这个:https://github.com/pydata/xarray/issues/2710