本文整理汇总了Python中xarray.DataArray.get_axis_num方法的典型用法代码示例。如果您正苦于以下问题:Python DataArray.get_axis_num方法的具体用法?Python DataArray.get_axis_num怎么用?Python DataArray.get_axis_num使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类xarray.DataArray
的用法示例。
在下文中一共展示了DataArray.get_axis_num方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: centered_to_right
# 需要导入模块: from xarray import DataArray [as 别名]
# 或者: from xarray.DataArray import get_axis_num [as 别名]
def centered_to_right(f: xr.DataArray, block_size, dim, boundary='wrap'):
"""Move centered variable to the right interface
Parameters
----------
f : xr.DataArray
block_size : size of the coarse graining block
dim : str
boundary : str, optional
A boundary condition which is passed to `isel_bc`
Returns
-------
interface : xr.DataArray
The value of f along the right interfaces of the coarse-grain blocks
"""
new_coord = get_center_coords(f[dim].values, block_size)
n = f.shape[f.get_axis_num(dim)]
left_idx = slice(block_size, n+1, block_size)
right_idx = slice(block_size-1, n, block_size)
left = isel_bc(f, left_idx, dim, boundary=boundary)
left = left.assign_coords(**{dim: new_coord})
right = isel_bc(f, right_idx, dim, boundary=boundary)
right = right.assign_coords(**{dim: new_coord})
return (left+right)/2