本文整理汇总了Python中xarray.DataArray.sel方法的典型用法代码示例。如果您正苦于以下问题:Python DataArray.sel方法的具体用法?Python DataArray.sel怎么用?Python DataArray.sel使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类xarray.DataArray
的用法示例。
在下文中一共展示了DataArray.sel方法的2个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: unstack_cat
# 需要导入模块: from xarray import DataArray [as 别名]
# 或者: from xarray.DataArray import sel [as 别名]
def unstack_cat(da: xr.DataArray, dim, level=0):
"""Unstack DataArray expanding to dataset along a given level
Parameters
----------
da
dim
level
Returns
-------
xr.Dataset
"""
if not isinstance(da, xr.DataArray):
raise ValueError("da must be a DataArray object")
idx = da.indexes[dim]
if not isinstance(idx, pd.MultiIndex):
raise ValueError(f"{dim} is not a stacked coordinate")
variables = idx.levels[level]
# pull variables out of datarray
data_dict = {}
for k in variables:
data_dict[k] = da.sel(variable=k).squeeze(drop=True)
# unstacked dataset
return xr.Dataset(data_dict)
示例2: ect_regional_subset
# 需要导入模块: from xarray import DataArray [as 别名]
# 或者: from xarray.DataArray import sel [as 别名]
def ect_regional_subset(da: xr.DataArray, lat_slice: slice, lon_slice: slice) -> xr.DataArray:
new_lat_slice = match_slice_order_to_index(da.coords['lat'], lat_slice)
new_lon_slice = match_slice_order_to_index(da.coords['lon'], lon_slice)
return da.sel(lat=new_lat_slice, lon=new_lon_slice)