Skip to content

get_dataset_dim_groups

Functions:

Name Description
get_dataset_dim_groups

Functions#

get_dataset_dim_groups #

get_dataset_dim_groups(
    dataset: xr.Dataset,
) -> dict[tuple[str, ...], list[str]]
Source code in tsdat/utils/get_dataset_dim_groups.py 
 6
 7
 8
 9
10
11
12
def get_dataset_dim_groups(dataset: xr.Dataset) -> dict[tuple[str, ...], list[str]]:
    dim_groups: dict[tuple[str, ...], list[str]] = defaultdict(list)
    for var_name, data_var in dataset.data_vars.items():
        dims = tuple(str(d) for d in data_var.dims)
        dim_groups[dims].append(var_name)

    return dim_groups