extras.pyi 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134
  1. from _typeshed import Incomplete
  2. import numpy as np
  3. from numpy.lib._function_base_impl import average
  4. from numpy.lib._index_tricks_impl import AxisConcatenator
  5. from .core import MaskedArray, dot
  6. __all__ = [
  7. "apply_along_axis",
  8. "apply_over_axes",
  9. "atleast_1d",
  10. "atleast_2d",
  11. "atleast_3d",
  12. "average",
  13. "clump_masked",
  14. "clump_unmasked",
  15. "column_stack",
  16. "compress_cols",
  17. "compress_nd",
  18. "compress_rowcols",
  19. "compress_rows",
  20. "corrcoef",
  21. "count_masked",
  22. "cov",
  23. "diagflat",
  24. "dot",
  25. "dstack",
  26. "ediff1d",
  27. "flatnotmasked_contiguous",
  28. "flatnotmasked_edges",
  29. "hsplit",
  30. "hstack",
  31. "in1d",
  32. "intersect1d",
  33. "isin",
  34. "mask_cols",
  35. "mask_rowcols",
  36. "mask_rows",
  37. "masked_all",
  38. "masked_all_like",
  39. "median",
  40. "mr_",
  41. "ndenumerate",
  42. "notmasked_contiguous",
  43. "notmasked_edges",
  44. "polyfit",
  45. "row_stack",
  46. "setdiff1d",
  47. "setxor1d",
  48. "stack",
  49. "union1d",
  50. "unique",
  51. "vander",
  52. "vstack",
  53. ]
  54. def count_masked(arr, axis=...): ...
  55. def masked_all(shape, dtype = ...): ...
  56. def masked_all_like(arr): ...
  57. class _fromnxfunction:
  58. __name__: Incomplete
  59. __doc__: Incomplete
  60. def __init__(self, funcname) -> None: ...
  61. def getdoc(self): ...
  62. def __call__(self, *args, **params): ...
  63. class _fromnxfunction_single(_fromnxfunction):
  64. def __call__(self, x, *args, **params): ...
  65. class _fromnxfunction_seq(_fromnxfunction):
  66. def __call__(self, x, *args, **params): ...
  67. class _fromnxfunction_allargs(_fromnxfunction):
  68. def __call__(self, *args, **params): ...
  69. atleast_1d: _fromnxfunction_allargs
  70. atleast_2d: _fromnxfunction_allargs
  71. atleast_3d: _fromnxfunction_allargs
  72. vstack: _fromnxfunction_seq
  73. row_stack: _fromnxfunction_seq
  74. hstack: _fromnxfunction_seq
  75. column_stack: _fromnxfunction_seq
  76. dstack: _fromnxfunction_seq
  77. stack: _fromnxfunction_seq
  78. hsplit: _fromnxfunction_single
  79. diagflat: _fromnxfunction_single
  80. def apply_along_axis(func1d, axis, arr, *args, **kwargs): ...
  81. def apply_over_axes(func, a, axes): ...
  82. def median(a, axis=..., out=..., overwrite_input=..., keepdims=...): ...
  83. def compress_nd(x, axis=...): ...
  84. def compress_rowcols(x, axis=...): ...
  85. def compress_rows(a): ...
  86. def compress_cols(a): ...
  87. def mask_rows(a, axis = ...): ...
  88. def mask_cols(a, axis = ...): ...
  89. def ediff1d(arr, to_end=..., to_begin=...): ...
  90. def unique(ar1, return_index=..., return_inverse=...): ...
  91. def intersect1d(ar1, ar2, assume_unique=...): ...
  92. def setxor1d(ar1, ar2, assume_unique=...): ...
  93. def in1d(ar1, ar2, assume_unique=..., invert=...): ...
  94. def isin(element, test_elements, assume_unique=..., invert=...): ...
  95. def union1d(ar1, ar2): ...
  96. def setdiff1d(ar1, ar2, assume_unique=...): ...
  97. def cov(x, y=..., rowvar=..., bias=..., allow_masked=..., ddof=...): ...
  98. def corrcoef(x, y=..., rowvar=..., bias = ..., allow_masked=..., ddof = ...): ...
  99. class MAxisConcatenator(AxisConcatenator):
  100. @staticmethod
  101. def concatenate(arrays: Incomplete, axis: int = 0) -> Incomplete: ... # type: ignore[override] # pyright: ignore[reportIncompatibleMethodOverride]
  102. @classmethod
  103. def makemat(cls, arr: Incomplete) -> Incomplete: ... # type: ignore[override] # pyright: ignore[reportIncompatibleVariableOverride]
  104. class mr_class(MAxisConcatenator):
  105. def __init__(self) -> None: ...
  106. mr_: mr_class
  107. def ndenumerate(a, compressed=...): ...
  108. def flatnotmasked_edges(a): ...
  109. def notmasked_edges(a, axis=...): ...
  110. def flatnotmasked_contiguous(a): ...
  111. def notmasked_contiguous(a, axis=...): ...
  112. def clump_unmasked(a): ...
  113. def clump_masked(a): ...
  114. def vander(x, n=...): ...
  115. def polyfit(x, y, deg, rcond=..., full=..., w=..., cov=...): ...
  116. #
  117. def mask_rowcols(a: Incomplete, axis: Incomplete | None = None) -> MaskedArray[Incomplete, np.dtype[Incomplete]]: ...