def create_multi_index(arr):
'''From DataArray arr make a pandas.MultiIndex for the arr.coords
Parameters
----------
arr: xarray.DataArray
Returns
-------
index: pandas.MultiIndex instance with index names
taken from arr.dims and levels taken from arr.coords
Examples
--------
'''
np_arrs = tuple(getattr(arr, dim).values for dim in arr.dims)
index = pd.MultiIndex.from_product(np_arrs, names=arr.dims)
return index
评论列表
文章目录