[英]Expand dimensions xarray
是否有现有的方法或方法来扩展xarray.DataArray
对象的维度(和坐标)?
我想获得类似于np.expand_dims的东西,同时为新扩展的DataArray
定义一个新的维度和坐标变量。
使用DataArray.assign_coords()
我可以创建一个新的坐标变量,但数组本身不会使用新轴进行扩展。
在xarray v0.10.0中,我使用assign_coords()
和expand_dims()
来添加新的维度和坐标变量。
例如:
import xarray as xr
import numpy as np
data = xr.DataArray([1, 2, 3], dims='x', coords={'x': [10, 20, 30]})
data_newcoord = data.assign_coords(y='coord_value')
data_expanded = data_newcoord.expand_dims('y')
print(data_expanded)
# <xarray.DataArray (y: 1, x: 3)>
# array([[1, 2, 3]])
# Coordinates:
# * x (x) int64 10 20 30
# * y (y) <U11 'coord_value'
我同意某种方法可以做到这一点。 它目前不存在于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'
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.