472 lines
14 KiB
Python
472 lines
14 KiB
Python
|
from collections.abc import Callable
|
||
|
|
from typing import Any, TypeVar
|
||
|
|
from numpy import ndarray, dtype, float64
|
||
|
|
|
||
|
|
from numpy import (
|
||
|
|
amax as amax,
|
||
|
|
amin as amin,
|
||
|
|
bool_ as bool_,
|
||
|
|
expand_dims as expand_dims,
|
||
|
|
clip as clip,
|
||
|
|
indices as indices,
|
||
|
|
ones_like as ones_like,
|
||
|
|
squeeze as squeeze,
|
||
|
|
zeros_like as zeros_like,
|
||
|
|
)
|
||
|
|
|
||
|
|
from numpy.lib.function_base import (
|
||
|
|
angle as angle,
|
||
|
|
)
|
||
|
|
|
||
|
|
# TODO: Set the `bound` to something more suitable once we
|
||
|
|
# have proper shape support
|
||
|
|
_ShapeType = TypeVar("_ShapeType", bound=Any)
|
||
|
|
_DType_co = TypeVar("_DType_co", bound=dtype[Any], covariant=True)
|
||
|
|
|
||
|
|
__all__: list[str]
|
||
|
|
|
||
|
|
MaskType = bool_
|
||
|
|
nomask: bool_
|
||
|
|
|
||
|
|
class MaskedArrayFutureWarning(FutureWarning): ...
|
||
|
|
class MAError(Exception): ...
|
||
|
|
class MaskError(MAError): ...
|
||
|
|
|
||
|
|
def default_fill_value(obj): ...
|
||
|
|
def minimum_fill_value(obj): ...
|
||
|
|
def maximum_fill_value(obj): ...
|
||
|
|
def set_fill_value(a, fill_value): ...
|
||
|
|
def common_fill_value(a, b): ...
|
||
|
|
def filled(a, fill_value=...): ...
|
||
|
|
def getdata(a, subok=...): ...
|
||
|
|
get_data = getdata
|
||
|
|
|
||
|
|
def fix_invalid(a, mask=..., copy=..., fill_value=...): ...
|
||
|
|
|
||
|
|
class _MaskedUFunc:
|
||
|
|
f: Any
|
||
|
|
__doc__: Any
|
||
|
|
__name__: Any
|
||
|
|
def __init__(self, ufunc): ...
|
||
|
|
|
||
|
|
class _MaskedUnaryOperation(_MaskedUFunc):
|
||
|
|
fill: Any
|
||
|
|
domain: Any
|
||
|
|
def __init__(self, mufunc, fill=..., domain=...): ...
|
||
|
|
def __call__(self, a, *args, **kwargs): ...
|
||
|
|
|
||
|
|
class _MaskedBinaryOperation(_MaskedUFunc):
|
||
|
|
fillx: Any
|
||
|
|
filly: Any
|
||
|
|
def __init__(self, mbfunc, fillx=..., filly=...): ...
|
||
|
|
def __call__(self, a, b, *args, **kwargs): ...
|
||
|
|
def reduce(self, target, axis=..., dtype=...): ...
|
||
|
|
def outer(self, a, b): ...
|
||
|
|
def accumulate(self, target, axis=...): ...
|
||
|
|
|
||
|
|
class _DomainedBinaryOperation(_MaskedUFunc):
|
||
|
|
domain: Any
|
||
|
|
fillx: Any
|
||
|
|
filly: Any
|
||
|
|
def __init__(self, dbfunc, domain, fillx=..., filly=...): ...
|
||
|
|
def __call__(self, a, b, *args, **kwargs): ...
|
||
|
|
|
||
|
|
exp: _MaskedUnaryOperation
|
||
|
|
conjugate: _MaskedUnaryOperation
|
||
|
|
sin: _MaskedUnaryOperation
|
||
|
|
cos: _MaskedUnaryOperation
|
||
|
|
arctan: _MaskedUnaryOperation
|
||
|
|
arcsinh: _MaskedUnaryOperation
|
||
|
|
sinh: _MaskedUnaryOperation
|
||
|
|
cosh: _MaskedUnaryOperation
|
||
|
|
tanh: _MaskedUnaryOperation
|
||
|
|
abs: _MaskedUnaryOperation
|
||
|
|
absolute: _MaskedUnaryOperation
|
||
|
|
fabs: _MaskedUnaryOperation
|
||
|
|
negative: _MaskedUnaryOperation
|
||
|
|
floor: _MaskedUnaryOperation
|
||
|
|
ceil: _MaskedUnaryOperation
|
||
|
|
around: _MaskedUnaryOperation
|
||
|
|
logical_not: _MaskedUnaryOperation
|
||
|
|
sqrt: _MaskedUnaryOperation
|
||
|
|
log: _MaskedUnaryOperation
|
||
|
|
log2: _MaskedUnaryOperation
|
||
|
|
log10: _MaskedUnaryOperation
|
||
|
|
tan: _MaskedUnaryOperation
|
||
|
|
arcsin: _MaskedUnaryOperation
|
||
|
|
arccos: _MaskedUnaryOperation
|
||
|
|
arccosh: _MaskedUnaryOperation
|
||
|
|
arctanh: _MaskedUnaryOperation
|
||
|
|
|
||
|
|
add: _MaskedBinaryOperation
|
||
|
|
subtract: _MaskedBinaryOperation
|
||
|
|
multiply: _MaskedBinaryOperation
|
||
|
|
arctan2: _MaskedBinaryOperation
|
||
|
|
equal: _MaskedBinaryOperation
|
||
|
|
not_equal: _MaskedBinaryOperation
|
||
|
|
less_equal: _MaskedBinaryOperation
|
||
|
|
greater_equal: _MaskedBinaryOperation
|
||
|
|
less: _MaskedBinaryOperation
|
||
|
|
greater: _MaskedBinaryOperation
|
||
|
|
logical_and: _MaskedBinaryOperation
|
||
|
|
alltrue: _MaskedBinaryOperation
|
||
|
|
logical_or: _MaskedBinaryOperation
|
||
|
|
sometrue: Callable[..., Any]
|
||
|
|
logical_xor: _MaskedBinaryOperation
|
||
|
|
bitwise_and: _MaskedBinaryOperation
|
||
|
|
bitwise_or: _MaskedBinaryOperation
|
||
|
|
bitwise_xor: _MaskedBinaryOperation
|
||
|
|
hypot: _MaskedBinaryOperation
|
||
|
|
divide: _MaskedBinaryOperation
|
||
|
|
true_divide: _MaskedBinaryOperation
|
||
|
|
floor_divide: _MaskedBinaryOperation
|
||
|
|
remainder: _MaskedBinaryOperation
|
||
|
|
fmod: _MaskedBinaryOperation
|
||
|
|
mod: _MaskedBinaryOperation
|
||
|
|
|
||
|
|
def make_mask_descr(ndtype): ...
|
||
|
|
def getmask(a): ...
|
||
|
|
get_mask = getmask
|
||
|
|
|
||
|
|
def getmaskarray(arr): ...
|
||
|
|
def is_mask(m): ...
|
||
|
|
def make_mask(m, copy=..., shrink=..., dtype=...): ...
|
||
|
|
def make_mask_none(newshape, dtype=...): ...
|
||
|
|
def mask_or(m1, m2, copy=..., shrink=...): ...
|
||
|
|
def flatten_mask(mask): ...
|
||
|
|
def masked_where(condition, a, copy=...): ...
|
||
|
|
def masked_greater(x, value, copy=...): ...
|
||
|
|
def masked_greater_equal(x, value, copy=...): ...
|
||
|
|
def masked_less(x, value, copy=...): ...
|
||
|
|
def masked_less_equal(x, value, copy=...): ...
|
||
|
|
def masked_not_equal(x, value, copy=...): ...
|
||
|
|
def masked_equal(x, value, copy=...): ...
|
||
|
|
def masked_inside(x, v1, v2, copy=...): ...
|
||
|
|
def masked_outside(x, v1, v2, copy=...): ...
|
||
|
|
def masked_object(x, value, copy=..., shrink=...): ...
|
||
|
|
def masked_values(x, value, rtol=..., atol=..., copy=..., shrink=...): ...
|
||
|
|
def masked_invalid(a, copy=...): ...
|
||
|
|
|
||
|
|
class _MaskedPrintOption:
|
||
|
|
def __init__(self, display): ...
|
||
|
|
def display(self): ...
|
||
|
|
def set_display(self, s): ...
|
||
|
|
def enabled(self): ...
|
||
|
|
def enable(self, shrink=...): ...
|
||
|
|
|
||
|
|
masked_print_option: _MaskedPrintOption
|
||
|
|
|
||
|
|
def flatten_structured_array(a): ...
|
||
|
|
|
||
|
|
class MaskedIterator:
|
||
|
|
ma: Any
|
||
|
|
dataiter: Any
|
||
|
|
maskiter: Any
|
||
|
|
def __init__(self, ma): ...
|
||
|
|
def __iter__(self): ...
|
||
|
|
def __getitem__(self, indx): ...
|
||
|
|
def __setitem__(self, index, value): ...
|
||
|
|
def __next__(self): ...
|
||
|
|
|
||
|
|
class MaskedArray(ndarray[_ShapeType, _DType_co]):
|
||
|
|
__array_priority__: Any
|
||
|
|
def __new__(cls, data=..., mask=..., dtype=..., copy=..., subok=..., ndmin=..., fill_value=..., keep_mask=..., hard_mask=..., shrink=..., order=...): ...
|
||
|
|
def __array_finalize__(self, obj): ...
|
||
|
|
def __array_wrap__(self, obj, context=...): ...
|
||
|
|
def view(self, dtype=..., type=..., fill_value=...): ...
|
||
|
|
def __getitem__(self, indx): ...
|
||
|
|
def __setitem__(self, indx, value): ...
|
||
|
|
@property
|
||
|
|
def dtype(self): ...
|
||
|
|
@dtype.setter
|
||
|
|
def dtype(self, dtype): ...
|
||
|
|
@property
|
||
|
|
def shape(self): ...
|
||
|
|
@shape.setter
|
||
|
|
def shape(self, shape): ...
|
||
|
|
def __setmask__(self, mask, copy=...): ...
|
||
|
|
@property
|
||
|
|
def mask(self): ...
|
||
|
|
@mask.setter
|
||
|
|
def mask(self, value): ...
|
||
|
|
@property
|
||
|
|
def recordmask(self): ...
|
||
|
|
@recordmask.setter
|
||
|
|
def recordmask(self, mask): ...
|
||
|
|
def harden_mask(self): ...
|
||
|
|
def soften_mask(self): ...
|
||
|
|
@property
|
||
|
|
def hardmask(self): ...
|
||
|
|
def unshare_mask(self): ...
|
||
|
|
@property
|
||
|
|
def sharedmask(self): ...
|
||
|
|
def shrink_mask(self): ...
|
||
|
|
@property
|
||
|
|
def baseclass(self): ...
|
||
|
|
data: Any
|
||
|
|
@property
|
||
|
|
def flat(self): ...
|
||
|
|
@flat.setter
|
||
|
|
def flat(self, value): ...
|
||
|
|
@property
|
||
|
|
def fill_value(self): ...
|
||
|
|
@fill_value.setter
|
||
|
|
def fill_value(self, value=...): ...
|
||
|
|
get_fill_value: Any
|
||
|
|
set_fill_value: Any
|
||
|
|
def filled(self, fill_value=...): ...
|
||
|
|
def compressed(self): ...
|
||
|
|
def compress(self, condition, axis=..., out=...): ...
|
||
|
|
def __eq__(self, other): ...
|
||
|
|
def __ne__(self, other): ...
|
||
|
|
def __ge__(self, other): ...
|
||
|
|
def __gt__(self, other): ...
|
||
|
|
def __le__(self, other): ...
|
||
|
|
def __lt__(self, other): ...
|
||
|
|
def __add__(self, other): ...
|
||
|
|
def __radd__(self, other): ...
|
||
|
|
def __sub__(self, other): ...
|
||
|
|
def __rsub__(self, other): ...
|
||
|
|
def __mul__(self, other): ...
|
||
|
|
def __rmul__(self, other): ...
|
||
|
|
def __div__(self, other): ...
|
||
|
|
def __truediv__(self, other): ...
|
||
|
|
def __rtruediv__(self, other): ...
|
||
|
|
def __floordiv__(self, other): ...
|
||
|
|
def __rfloordiv__(self, other): ...
|
||
|
|
def __pow__(self, other): ...
|
||
|
|
def __rpow__(self, other): ...
|
||
|
|
def __iadd__(self, other): ...
|
||
|
|
def __isub__(self, other): ...
|
||
|
|
def __imul__(self, other): ...
|
||
|
|
def __idiv__(self, other): ...
|
||
|
|
def __ifloordiv__(self, other): ...
|
||
|
|
def __itruediv__(self, other): ...
|
||
|
|
def __ipow__(self, other): ...
|
||
|
|
def __float__(self): ...
|
||
|
|
def __int__(self): ...
|
||
|
|
@property # type: ignore[misc]
|
||
|
|
def imag(self): ...
|
||
|
|
get_imag: Any
|
||
|
|
@property # type: ignore[misc]
|
||
|
|
def real(self): ...
|
||
|
|
get_real: Any
|
||
|
|
def count(self, axis=..., keepdims=...): ...
|
||
|
|
def ravel(self, order=...): ...
|
||
|
|
def reshape(self, *s, **kwargs): ...
|
||
|
|
def resize(self, newshape, refcheck=..., order=...): ...
|
||
|
|
def put(self, indices, values, mode=...): ...
|
||
|
|
def ids(self): ...
|
||
|
|
def iscontiguous(self): ...
|
||
|
|
def all(self, axis=..., out=..., keepdims=...): ...
|
||
|
|
def any(self, axis=..., out=..., keepdims=...): ...
|
||
|
|
def nonzero(self): ...
|
||
|
|
def trace(self, offset=..., axis1=..., axis2=..., dtype=..., out=...): ...
|
||
|
|
def dot(self, b, out=..., strict=...): ...
|
||
|
|
def sum(self, axis=..., dtype=..., out=..., keepdims=...): ...
|
||
|
|
def cumsum(self, axis=..., dtype=..., out=...): ...
|
||
|
|
def prod(self, axis=..., dtype=..., out=..., keepdims=...): ...
|
||
|
|
product: Any
|
||
|
|
def cumprod(self, axis=..., dtype=..., out=...): ...
|
||
|
|
def mean(self, axis=..., dtype=..., out=..., keepdims=...): ...
|
||
|
|
def anom(self, axis=..., dtype=...): ...
|
||
|
|
def var(self, axis=..., dtype=..., out=..., ddof=..., keepdims=...): ...
|
||
|
|
def std(self, axis=..., dtype=..., out=..., ddof=..., keepdims=...): ...
|
||
|
|
def round(self, decimals=..., out=...): ...
|
||
|
|
def argsort(self, axis=..., kind=..., order=..., endwith=..., fill_value=...): ...
|
||
|
|
def argmin(self, axis=..., fill_value=..., out=..., *, keepdims=...): ...
|
||
|
|
def argmax(self, axis=..., fill_value=..., out=..., *, keepdims=...): ...
|
||
|
|
def sort(self, axis=..., kind=..., order=..., endwith=..., fill_value=...): ...
|
||
|
|
def min(self, axis=..., out=..., fill_value=..., keepdims=...): ...
|
||
|
|
# NOTE: deprecated
|
||
|
|
# def tostring(self, fill_value=..., order=...): ...
|
||
|
|
def max(self, axis=..., out=..., fill_value=..., keepdims=...): ...
|
||
|
|
def ptp(self, axis=..., out=..., fill_value=..., keepdims=...): ...
|
||
|
|
def partition(self, *args, **kwargs): ...
|
||
|
|
def argpartition(self, *args, **kwargs): ...
|
||
|
|
def take(self, indices, axis=..., out=..., mode=...): ...
|
||
|
|
copy: Any
|
||
|
|
diagonal: Any
|
||
|
|
flatten: Any
|
||
|
|
repeat: Any
|
||
|
|
squeeze: Any
|
||
|
|
swapaxes: Any
|
||
|
|
T: Any
|
||
|
|
transpose: Any
|
||
|
|
def tolist(self, fill_value=...): ...
|
||
|
|
def tobytes(self, fill_value=..., order=...): ...
|
||
|
|
def tofile(self, fid, sep=..., format=...): ...
|
||
|
|
def toflex(self): ...
|
||
|
|
torecords: Any
|
||
|
|
def __reduce__(self): ...
|
||
|
|
def __deepcopy__(self, memo=...): ...
|
||
|
|
|
||
|
|
class mvoid(MaskedArray[_ShapeType, _DType_co]):
|
||
|
|
def __new__(
|
||
|
|
self,
|
||
|
|
data,
|
||
|
|
mask=...,
|
||
|
|
dtype=...,
|
||
|
|
fill_value=...,
|
||
|
|
hardmask=...,
|
||
|
|
copy=...,
|
||
|
|
subok=...,
|
||
|
|
): ...
|
||
|
|
def __getitem__(self, indx): ...
|
||
|
|
def __setitem__(self, indx, value): ...
|
||
|
|
def __iter__(self): ...
|
||
|
|
def __len__(self): ...
|
||
|
|
def filled(self, fill_value=...): ...
|
||
|
|
def tolist(self): ...
|
||
|
|
|
||
|
|
def isMaskedArray(x): ...
|
||
|
|
isarray = isMaskedArray
|
||
|
|
isMA = isMaskedArray
|
||
|
|
|
||
|
|
# 0D float64 array
|
||
|
|
class MaskedConstant(MaskedArray[Any, dtype[float64]]):
|
||
|
|
def __new__(cls): ...
|
||
|
|
__class__: Any
|
||
|
|
def __array_finalize__(self, obj): ...
|
||
|
|
def __array_prepare__(self, obj, context=...): ...
|
||
|
|
def __array_wrap__(self, obj, context=...): ...
|
||
|
|
def __format__(self, format_spec): ...
|
||
|
|
def __reduce__(self): ...
|
||
|
|
def __iop__(self, other): ...
|
||
|
|
__iadd__: Any
|
||
|
|
__isub__: Any
|
||
|
|
__imul__: Any
|
||
|
|
__ifloordiv__: Any
|
||
|
|
__itruediv__: Any
|
||
|
|
__ipow__: Any
|
||
|
|
def copy(self, *args, **kwargs): ...
|
||
|
|
def __copy__(self): ...
|
||
|
|
def __deepcopy__(self, memo): ...
|
||
|
|
def __setattr__(self, attr, value): ...
|
||
|
|
|
||
|
|
masked: MaskedConstant
|
||
|
|
masked_singleton: MaskedConstant
|
||
|
|
masked_array = MaskedArray
|
||
|
|
|
||
|
|
def array(
|
||
|
|
data,
|
||
|
|
dtype=...,
|
||
|
|
copy=...,
|
||
|
|
order=...,
|
||
|
|
mask=...,
|
||
|
|
fill_value=...,
|
||
|
|
keep_mask=...,
|
||
|
|
hard_mask=...,
|
||
|
|
shrink=...,
|
||
|
|
subok=...,
|
||
|
|
ndmin=...,
|
||
|
|
): ...
|
||
|
|
def is_masked(x): ...
|
||
|
|
|
||
|
|
class _extrema_operation(_MaskedUFunc):
|
||
|
|
compare: Any
|
||
|
|
fill_value_func: Any
|
||
|
|
def __init__(self, ufunc, compare, fill_value): ...
|
||
|
|
# NOTE: in practice `b` has a default value, but users should
|
||
|
|
# explicitly provide a value here as the default is deprecated
|
||
|
|
def __call__(self, a, b): ...
|
||
|
|
def reduce(self, target, axis=...): ...
|
||
|
|
def outer(self, a, b): ...
|
||
|
|
|
||
|
|
def min(obj, axis=..., out=..., fill_value=..., keepdims=...): ...
|
||
|
|
def max(obj, axis=..., out=..., fill_value=..., keepdims=...): ...
|
||
|
|
def ptp(obj, axis=..., out=..., fill_value=..., keepdims=...): ...
|
||
|
|
|
||
|
|
class _frommethod:
|
||
|
|
__name__: Any
|
||
|
|
__doc__: Any
|
||
|
|
reversed: Any
|
||
|
|
def __init__(self, methodname, reversed=...): ...
|
||
|
|
def getdoc(self): ...
|
||
|
|
def __call__(self, a, *args, **params): ...
|
||
|
|
|
||
|
|
all: _frommethod
|
||
|
|
anomalies: _frommethod
|
||
|
|
anom: _frommethod
|
||
|
|
any: _frommethod
|
||
|
|
compress: _frommethod
|
||
|
|
cumprod: _frommethod
|
||
|
|
cumsum: _frommethod
|
||
|
|
copy: _frommethod
|
||
|
|
diagonal: _frommethod
|
||
|
|
harden_mask: _frommethod
|
||
|
|
ids: _frommethod
|
||
|
|
mean: _frommethod
|
||
|
|
nonzero: _frommethod
|
||
|
|
prod: _frommethod
|
||
|
|
product: _frommethod
|
||
|
|
ravel: _frommethod
|
||
|
|
repeat: _frommethod
|
||
|
|
soften_mask: _frommethod
|
||
|
|
std: _frommethod
|
||
|
|
sum: _frommethod
|
||
|
|
swapaxes: _frommethod
|
||
|
|
trace: _frommethod
|
||
|
|
var: _frommethod
|
||
|
|
count: _frommethod
|
||
|
|
argmin: _frommethod
|
||
|
|
argmax: _frommethod
|
||
|
|
|
||
|
|
minimum: _extrema_operation
|
||
|
|
maximum: _extrema_operation
|
||
|
|
|
||
|
|
def take(a, indices, axis=..., out=..., mode=...): ...
|
||
|
|
def power(a, b, third=...): ...
|
||
|
|
def argsort(a, axis=..., kind=..., order=..., endwith=..., fill_value=...): ...
|
||
|
|
def sort(a, axis=..., kind=..., order=..., endwith=..., fill_value=...): ...
|
||
|
|
def compressed(x): ...
|
||
|
|
def concatenate(arrays, axis=...): ...
|
||
|
|
def diag(v, k=...): ...
|
||
|
|
def left_shift(a, n): ...
|
||
|
|
def right_shift(a, n): ...
|
||
|
|
def put(a, indices, values, mode=...): ...
|
||
|
|
def putmask(a, mask, values): ...
|
||
|
|
def transpose(a, axes=...): ...
|
||
|
|
def reshape(a, new_shape, order=...): ...
|
||
|
|
def resize(x, new_shape): ...
|
||
|
|
def ndim(obj): ...
|
||
|
|
def shape(obj): ...
|
||
|
|
def size(obj, axis=...): ...
|
||
|
|
def diff(a, /, n=..., axis=..., prepend=..., append=...): ...
|
||
|
|
def where(condition, x=..., y=...): ...
|
||
|
|
def choose(indices, choices, out=..., mode=...): ...
|
||
|
|
def round(a, decimals=..., out=...): ...
|
||
|
|
|
||
|
|
def inner(a, b): ...
|
||
|
|
innerproduct = inner
|
||
|
|
|
||
|
|
def outer(a, b): ...
|
||
|
|
outerproduct = outer
|
||
|
|
|
||
|
|
def correlate(a, v, mode=..., propagate_mask=...): ...
|
||
|
|
def convolve(a, v, mode=..., propagate_mask=...): ...
|
||
|
|
def allequal(a, b, fill_value=...): ...
|
||
|
|
def allclose(a, b, masked_equal=..., rtol=..., atol=...): ...
|
||
|
|
def asarray(a, dtype=..., order=...): ...
|
||
|
|
def asanyarray(a, dtype=...): ...
|
||
|
|
def fromflex(fxarray): ...
|
||
|
|
|
||
|
|
class _convert2ma:
|
||
|
|
__doc__: Any
|
||
|
|
def __init__(self, funcname, params=...): ...
|
||
|
|
def getdoc(self): ...
|
||
|
|
def __call__(self, *args, **params): ...
|
||
|
|
|
||
|
|
arange: _convert2ma
|
||
|
|
empty: _convert2ma
|
||
|
|
empty_like: _convert2ma
|
||
|
|
frombuffer: _convert2ma
|
||
|
|
fromfunction: _convert2ma
|
||
|
|
identity: _convert2ma
|
||
|
|
ones: _convert2ma
|
||
|
|
zeros: _convert2ma
|
||
|
|
|
||
|
|
def append(a, b, axis=...): ...
|
||
|
|
def dot(a, b, strict=..., out=...): ...
|
||
|
|
def mask_rowcols(a, axis=...): ...
|