xarray.DataTree.subtree_with_keys