earthkit.data.indexing.tensor

Classes

Functions

coords_to_index(coords, shape)

Map user coords to field index.

flatten_arg(func)

index_to_coords(index, shape)

Module Contents

class earthkit.data.indexing.tensor.CubeCoords

Bases: dict

dict() -> new empty dictionary dict(mapping) -> new dictionary initialized from a mapping object’s

(key, value) pairs

dict(iterable) -> new dictionary initialized as if via:

d = {} for k, v in iterable:

d[k] = v

dict(**kwargs) -> new dictionary initialized with the name=value pairs

in the keyword argument list. For example: dict(one=1, two=2)

class earthkit.data.indexing.tensor.CubeSelection(kwargs, remapping=None)

Bases: earthkit.data.core.index.Selection

actions
property is_empty
match_element(element)
remapping = None
class earthkit.data.indexing.tensor.FieldListSparseTensor(source, user_coords, field_coords, field_dims, flatten_values, user_coords_to_fl_idx)

Bases: FieldListTensor

clear()
property field_coords
property field_dims
field_indexes(indexes)
property field_shape
flatten_values
classmethod from_fieldlist(ds, *args, remapping=None, flatten_values=False, sort=True, progress_bar=False, user_dims_and_coords=None, field_dims_and_coords=None, allow_holes=False)
classmethod from_tensor(owner, source, user_coords, user_coords_to_fl_idx)
property full_coords
property full_dims
property full_shape
is_full_field(indexes)
isel(*args, remapping=None, **kwargs)
latitudes(**kwargs)
longitudes(**kwargs)
sel(*args, remapping=None, **kwargs)
source
to_array(dtype=None, array_namespace=None, device=None, index=None, **kwargs)
to_numpy(dtype=None, index=None, **kwargs)
property user_coords
property user_dims
property user_shape
class earthkit.data.indexing.tensor.FieldListTensor(source, user_coords, field_coords, field_dims, flatten_values, check_if_tensor_is_full=True)

Bases: TensorCore

clear()
property field_coords
property field_dims
field_indexes(indexes)
property field_shape
flatten_values
classmethod from_fieldlist(ds, *args, remapping=None, flatten_values=False, sort=True, progress_bar=False, user_dims_and_coords=None, field_dims_and_coords=None, allow_holes=False)
classmethod from_tensor(owner, source, user_coords)
property full_coords
property full_dims
property full_shape
is_full_field(indexes)
isel(*args, remapping=None, **kwargs)
latitudes(**kwargs)
longitudes(**kwargs)
sel(*args, remapping=None, **kwargs)
source
to_array(dtype=None, array_namespace=None, device=None, index=None, **kwargs)
to_numpy(dtype=None, index=None, **kwargs)
property user_coords
property user_dims
property user_shape
class earthkit.data.indexing.tensor.TensorCore
property field_coords
property field_dims
property field_shape
flatten_values = None
property full_coords
property full_dims
property full_shape
isel(*args, remapping=None, **kwargs)
sel(*args, remapping=None, **kwargs)
abstractmethod to_numpy(**kwargs)
property user_coords
property user_dims
property user_shape
earthkit.data.indexing.tensor.coords_to_index(coords, shape)

Map user coords to field index.

earthkit.data.indexing.tensor.flatten_arg(func)
earthkit.data.indexing.tensor.index_to_coords(index, shape)