neovim/typings/numpy/ma/core.pyi

"""
This type stub file was generated by pyright.
"""

from collections.abc import Callable
from typing import Any, TypeVar
from numpy import bool_ as bool_, dtype, float64, ndarray, squeeze as squeeze

_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 = ...
def fix_invalid(a, mask=..., copy=..., fill_value=...):
    ...

class _MaskedUFunc:
    f: Any
    __doc__: Any
    __name__: Any
    def __init__(self, ufunc) -> None:
        ...
    


class _MaskedUnaryOperation(_MaskedUFunc):
    fill: Any
    domain: Any
    def __init__(self, mufunc, fill=..., domain=...) -> None:
        ...
    
    def __call__(self, a, *args, **kwargs):
        ...
    


class _MaskedBinaryOperation(_MaskedUFunc):
    fillx: Any
    filly: Any
    def __init__(self, mbfunc, fillx=..., filly=...) -> None:
        ...
    
    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=...) -> None:
        ...
    
    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 = ...
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) -> None:
        ...
    
    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) -> None:
        ...
    
    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) -> bool:
        ...
    
    def __ne__(self, other) -> bool:
        ...
    
    def __ge__(self, other) -> bool:
        ...
    
    def __gt__(self, other) -> bool:
        ...
    
    def __le__(self, other) -> bool:
        ...
    
    def __lt__(self, other) -> bool:
        ...
    
    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) -> int:
        ...
    
    @property
    def imag(self):
        ...
    
    get_imag: Any
    @property
    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=...):
        ...
    
    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 = ...
isMA = ...
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) -> None:
        ...
    
    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=...) -> None:
        ...
    
    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 = ...
def outer(a, b):
    ...

outerproduct = ...
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=...) -> None:
        ...
    
    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=...):
    ...