Home Articles

将维度添加到xarray DataArray

Asked
Viewed 1925 times
6

我需要为 DataArray 添加维度,填充新维度的值 . 这是原始数组 .

a_size = 10
a_coords = np.linspace(0, 1, a_size)

b_size = 5
b_coords = np.linspace(0, 1, b_size)

# original 1-dimensional array
x = xr.DataArray(
    np.random.random(a_size),
    coords=[('a', a coords)])

我想我可以创建一个带有新维度的空DataArray并复制现有数据 .

y = xr.DataArray(
    np.empty((b_size, a_size),
    coords=([('b', b_coords), ('a', a_coords)])
y[:] = x

更好的想法可能是使用 concat . 我花了一段时间才弄清楚如何为concat维度指定dims和coords,并且这些选项都不是很好 . 有什么我想念的东西可以使这个版本更干净吗?

# specify the dimension name, then set the coordinates
y = xr.concat([x for _ in b_coords], 'b')
y['b'] = b_coords

# specify the coordinates, then rename the dimension
y = xr.concat([x for _ in b_coords], b_coords)
y.rename({'concat_dim': 'b'})

# use a DataArray as the concat dimension
y = xr.concat(
    [x for _ in b_coords],
    xr.DataArray(b_coords, name='b', dims=['b']))

不过,有没有比上述两个选项更好的方法呢?

2 Answers

  • 4

    由于数学应用于新维度的方式,我喜欢乘以以添加新维度 .

    identityb = xr.DataArray(np.ones_like(b_coords), coords=[('b', b_coords)])
    y = x * identityb
    
  • 3

    您已经对当前选项进行了非常彻底的分析,实际上这些选项都不是很干净 .

    这肯定是为xarray编写的有用功能,但是还没有人能够实现它 . 也许你会有兴趣帮忙吗?

    有关API提案,请参阅此GitHub问题:https://github.com/pydata/xarray/issues/170

Related