numpy 2.3.5__cp313-cp313-macosx_14_0_arm64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of numpy might be problematic. Click here for more details.

Files changed (897) hide show
  1. numpy/__config__.py +170 -0
  2. numpy/__config__.pyi +102 -0
  3. numpy/__init__.cython-30.pxd +1241 -0
  4. numpy/__init__.pxd +1154 -0
  5. numpy/__init__.py +945 -0
  6. numpy/__init__.pyi +6147 -0
  7. numpy/_array_api_info.py +346 -0
  8. numpy/_array_api_info.pyi +207 -0
  9. numpy/_configtool.py +39 -0
  10. numpy/_configtool.pyi +1 -0
  11. numpy/_core/__init__.py +186 -0
  12. numpy/_core/__init__.pyi +2 -0
  13. numpy/_core/_add_newdocs.py +6967 -0
  14. numpy/_core/_add_newdocs.pyi +3 -0
  15. numpy/_core/_add_newdocs_scalars.py +390 -0
  16. numpy/_core/_add_newdocs_scalars.pyi +16 -0
  17. numpy/_core/_asarray.py +134 -0
  18. numpy/_core/_asarray.pyi +41 -0
  19. numpy/_core/_dtype.py +366 -0
  20. numpy/_core/_dtype.pyi +58 -0
  21. numpy/_core/_dtype_ctypes.py +120 -0
  22. numpy/_core/_dtype_ctypes.pyi +83 -0
  23. numpy/_core/_exceptions.py +162 -0
  24. numpy/_core/_exceptions.pyi +55 -0
  25. numpy/_core/_internal.py +958 -0
  26. numpy/_core/_internal.pyi +72 -0
  27. numpy/_core/_machar.py +355 -0
  28. numpy/_core/_machar.pyi +55 -0
  29. numpy/_core/_methods.py +255 -0
  30. numpy/_core/_methods.pyi +22 -0
  31. numpy/_core/_multiarray_tests.cpython-313-darwin.so +0 -0
  32. numpy/_core/_multiarray_umath.cpython-313-darwin.so +0 -0
  33. numpy/_core/_operand_flag_tests.cpython-313-darwin.so +0 -0
  34. numpy/_core/_rational_tests.cpython-313-darwin.so +0 -0
  35. numpy/_core/_simd.cpython-313-darwin.so +0 -0
  36. numpy/_core/_simd.pyi +25 -0
  37. numpy/_core/_string_helpers.py +100 -0
  38. numpy/_core/_string_helpers.pyi +12 -0
  39. numpy/_core/_struct_ufunc_tests.cpython-313-darwin.so +0 -0
  40. numpy/_core/_type_aliases.py +119 -0
  41. numpy/_core/_type_aliases.pyi +97 -0
  42. numpy/_core/_ufunc_config.py +491 -0
  43. numpy/_core/_ufunc_config.pyi +78 -0
  44. numpy/_core/_umath_tests.cpython-313-darwin.so +0 -0
  45. numpy/_core/arrayprint.py +1775 -0
  46. numpy/_core/arrayprint.pyi +238 -0
  47. numpy/_core/cversions.py +13 -0
  48. numpy/_core/defchararray.py +1427 -0
  49. numpy/_core/defchararray.pyi +1135 -0
  50. numpy/_core/einsumfunc.py +1498 -0
  51. numpy/_core/einsumfunc.pyi +184 -0
  52. numpy/_core/fromnumeric.py +4269 -0
  53. numpy/_core/fromnumeric.pyi +1750 -0
  54. numpy/_core/function_base.py +545 -0
  55. numpy/_core/function_base.pyi +278 -0
  56. numpy/_core/getlimits.py +748 -0
  57. numpy/_core/getlimits.pyi +3 -0
  58. numpy/_core/include/numpy/__multiarray_api.c +376 -0
  59. numpy/_core/include/numpy/__multiarray_api.h +1628 -0
  60. numpy/_core/include/numpy/__ufunc_api.c +54 -0
  61. numpy/_core/include/numpy/__ufunc_api.h +341 -0
  62. numpy/_core/include/numpy/_neighborhood_iterator_imp.h +90 -0
  63. numpy/_core/include/numpy/_numpyconfig.h +33 -0
  64. numpy/_core/include/numpy/_public_dtype_api_table.h +86 -0
  65. numpy/_core/include/numpy/arrayobject.h +7 -0
  66. numpy/_core/include/numpy/arrayscalars.h +196 -0
  67. numpy/_core/include/numpy/dtype_api.h +480 -0
  68. numpy/_core/include/numpy/halffloat.h +70 -0
  69. numpy/_core/include/numpy/ndarrayobject.h +304 -0
  70. numpy/_core/include/numpy/ndarraytypes.h +1950 -0
  71. numpy/_core/include/numpy/npy_2_compat.h +249 -0
  72. numpy/_core/include/numpy/npy_2_complexcompat.h +28 -0
  73. numpy/_core/include/numpy/npy_3kcompat.h +374 -0
  74. numpy/_core/include/numpy/npy_common.h +977 -0
  75. numpy/_core/include/numpy/npy_cpu.h +124 -0
  76. numpy/_core/include/numpy/npy_endian.h +78 -0
  77. numpy/_core/include/numpy/npy_math.h +602 -0
  78. numpy/_core/include/numpy/npy_no_deprecated_api.h +20 -0
  79. numpy/_core/include/numpy/npy_os.h +42 -0
  80. numpy/_core/include/numpy/numpyconfig.h +182 -0
  81. numpy/_core/include/numpy/random/LICENSE.txt +21 -0
  82. numpy/_core/include/numpy/random/bitgen.h +20 -0
  83. numpy/_core/include/numpy/random/distributions.h +209 -0
  84. numpy/_core/include/numpy/random/libdivide.h +2079 -0
  85. numpy/_core/include/numpy/ufuncobject.h +343 -0
  86. numpy/_core/include/numpy/utils.h +37 -0
  87. numpy/_core/lib/libnpymath.a +0 -0
  88. numpy/_core/lib/npy-pkg-config/mlib.ini +12 -0
  89. numpy/_core/lib/npy-pkg-config/npymath.ini +20 -0
  90. numpy/_core/lib/pkgconfig/numpy.pc +7 -0
  91. numpy/_core/memmap.py +363 -0
  92. numpy/_core/memmap.pyi +3 -0
  93. numpy/_core/multiarray.py +1762 -0
  94. numpy/_core/multiarray.pyi +1285 -0
  95. numpy/_core/numeric.py +2760 -0
  96. numpy/_core/numeric.pyi +882 -0
  97. numpy/_core/numerictypes.py +633 -0
  98. numpy/_core/numerictypes.pyi +197 -0
  99. numpy/_core/overrides.py +183 -0
  100. numpy/_core/overrides.pyi +48 -0
  101. numpy/_core/printoptions.py +32 -0
  102. numpy/_core/printoptions.pyi +28 -0
  103. numpy/_core/records.py +1089 -0
  104. numpy/_core/records.pyi +333 -0
  105. numpy/_core/shape_base.py +998 -0
  106. numpy/_core/shape_base.pyi +175 -0
  107. numpy/_core/strings.py +1829 -0
  108. numpy/_core/strings.pyi +511 -0
  109. numpy/_core/tests/_locales.py +72 -0
  110. numpy/_core/tests/_natype.py +205 -0
  111. numpy/_core/tests/data/astype_copy.pkl +0 -0
  112. numpy/_core/tests/data/generate_umath_validation_data.cpp +170 -0
  113. numpy/_core/tests/data/recarray_from_file.fits +0 -0
  114. numpy/_core/tests/data/umath-validation-set-README.txt +15 -0
  115. numpy/_core/tests/data/umath-validation-set-arccos.csv +1429 -0
  116. numpy/_core/tests/data/umath-validation-set-arccosh.csv +1429 -0
  117. numpy/_core/tests/data/umath-validation-set-arcsin.csv +1429 -0
  118. numpy/_core/tests/data/umath-validation-set-arcsinh.csv +1429 -0
  119. numpy/_core/tests/data/umath-validation-set-arctan.csv +1429 -0
  120. numpy/_core/tests/data/umath-validation-set-arctanh.csv +1429 -0
  121. numpy/_core/tests/data/umath-validation-set-cbrt.csv +1429 -0
  122. numpy/_core/tests/data/umath-validation-set-cos.csv +1375 -0
  123. numpy/_core/tests/data/umath-validation-set-cosh.csv +1429 -0
  124. numpy/_core/tests/data/umath-validation-set-exp.csv +412 -0
  125. numpy/_core/tests/data/umath-validation-set-exp2.csv +1429 -0
  126. numpy/_core/tests/data/umath-validation-set-expm1.csv +1429 -0
  127. numpy/_core/tests/data/umath-validation-set-log.csv +271 -0
  128. numpy/_core/tests/data/umath-validation-set-log10.csv +1629 -0
  129. numpy/_core/tests/data/umath-validation-set-log1p.csv +1429 -0
  130. numpy/_core/tests/data/umath-validation-set-log2.csv +1629 -0
  131. numpy/_core/tests/data/umath-validation-set-sin.csv +1370 -0
  132. numpy/_core/tests/data/umath-validation-set-sinh.csv +1429 -0
  133. numpy/_core/tests/data/umath-validation-set-tan.csv +1429 -0
  134. numpy/_core/tests/data/umath-validation-set-tanh.csv +1429 -0
  135. numpy/_core/tests/examples/cython/checks.pyx +373 -0
  136. numpy/_core/tests/examples/cython/meson.build +43 -0
  137. numpy/_core/tests/examples/cython/setup.py +39 -0
  138. numpy/_core/tests/examples/limited_api/limited_api1.c +17 -0
  139. numpy/_core/tests/examples/limited_api/limited_api2.pyx +11 -0
  140. numpy/_core/tests/examples/limited_api/limited_api_latest.c +19 -0
  141. numpy/_core/tests/examples/limited_api/meson.build +59 -0
  142. numpy/_core/tests/examples/limited_api/setup.py +24 -0
  143. numpy/_core/tests/test__exceptions.py +90 -0
  144. numpy/_core/tests/test_abc.py +54 -0
  145. numpy/_core/tests/test_api.py +654 -0
  146. numpy/_core/tests/test_argparse.py +92 -0
  147. numpy/_core/tests/test_array_api_info.py +113 -0
  148. numpy/_core/tests/test_array_coercion.py +911 -0
  149. numpy/_core/tests/test_array_interface.py +222 -0
  150. numpy/_core/tests/test_arraymethod.py +84 -0
  151. numpy/_core/tests/test_arrayobject.py +75 -0
  152. numpy/_core/tests/test_arrayprint.py +1328 -0
  153. numpy/_core/tests/test_casting_floatingpoint_errors.py +154 -0
  154. numpy/_core/tests/test_casting_unittests.py +817 -0
  155. numpy/_core/tests/test_conversion_utils.py +206 -0
  156. numpy/_core/tests/test_cpu_dispatcher.py +49 -0
  157. numpy/_core/tests/test_cpu_features.py +432 -0
  158. numpy/_core/tests/test_custom_dtypes.py +315 -0
  159. numpy/_core/tests/test_cython.py +351 -0
  160. numpy/_core/tests/test_datetime.py +2734 -0
  161. numpy/_core/tests/test_defchararray.py +825 -0
  162. numpy/_core/tests/test_deprecations.py +454 -0
  163. numpy/_core/tests/test_dlpack.py +190 -0
  164. numpy/_core/tests/test_dtype.py +1995 -0
  165. numpy/_core/tests/test_einsum.py +1317 -0
  166. numpy/_core/tests/test_errstate.py +131 -0
  167. numpy/_core/tests/test_extint128.py +217 -0
  168. numpy/_core/tests/test_function_base.py +503 -0
  169. numpy/_core/tests/test_getlimits.py +205 -0
  170. numpy/_core/tests/test_half.py +568 -0
  171. numpy/_core/tests/test_hashtable.py +35 -0
  172. numpy/_core/tests/test_indexerrors.py +125 -0
  173. numpy/_core/tests/test_indexing.py +1455 -0
  174. numpy/_core/tests/test_item_selection.py +167 -0
  175. numpy/_core/tests/test_limited_api.py +102 -0
  176. numpy/_core/tests/test_longdouble.py +369 -0
  177. numpy/_core/tests/test_machar.py +30 -0
  178. numpy/_core/tests/test_mem_overlap.py +930 -0
  179. numpy/_core/tests/test_mem_policy.py +452 -0
  180. numpy/_core/tests/test_memmap.py +246 -0
  181. numpy/_core/tests/test_multiarray.py +10577 -0
  182. numpy/_core/tests/test_multithreading.py +292 -0
  183. numpy/_core/tests/test_nditer.py +3498 -0
  184. numpy/_core/tests/test_nep50_promotions.py +287 -0
  185. numpy/_core/tests/test_numeric.py +4247 -0
  186. numpy/_core/tests/test_numerictypes.py +651 -0
  187. numpy/_core/tests/test_overrides.py +791 -0
  188. numpy/_core/tests/test_print.py +200 -0
  189. numpy/_core/tests/test_protocols.py +46 -0
  190. numpy/_core/tests/test_records.py +544 -0
  191. numpy/_core/tests/test_regression.py +2670 -0
  192. numpy/_core/tests/test_scalar_ctors.py +207 -0
  193. numpy/_core/tests/test_scalar_methods.py +246 -0
  194. numpy/_core/tests/test_scalarbuffer.py +153 -0
  195. numpy/_core/tests/test_scalarinherit.py +105 -0
  196. numpy/_core/tests/test_scalarmath.py +1176 -0
  197. numpy/_core/tests/test_scalarprint.py +403 -0
  198. numpy/_core/tests/test_shape_base.py +891 -0
  199. numpy/_core/tests/test_simd.py +1341 -0
  200. numpy/_core/tests/test_simd_module.py +103 -0
  201. numpy/_core/tests/test_stringdtype.py +1814 -0
  202. numpy/_core/tests/test_strings.py +1499 -0
  203. numpy/_core/tests/test_ufunc.py +3313 -0
  204. numpy/_core/tests/test_umath.py +4928 -0
  205. numpy/_core/tests/test_umath_accuracy.py +124 -0
  206. numpy/_core/tests/test_umath_complex.py +626 -0
  207. numpy/_core/tests/test_unicode.py +368 -0
  208. numpy/_core/umath.py +60 -0
  209. numpy/_core/umath.pyi +197 -0
  210. numpy/_distributor_init.py +15 -0
  211. numpy/_distributor_init.pyi +1 -0
  212. numpy/_expired_attrs_2_0.py +79 -0
  213. numpy/_expired_attrs_2_0.pyi +62 -0
  214. numpy/_globals.py +96 -0
  215. numpy/_globals.pyi +17 -0
  216. numpy/_pyinstaller/__init__.py +0 -0
  217. numpy/_pyinstaller/__init__.pyi +0 -0
  218. numpy/_pyinstaller/hook-numpy.py +36 -0
  219. numpy/_pyinstaller/hook-numpy.pyi +13 -0
  220. numpy/_pyinstaller/tests/__init__.py +16 -0
  221. numpy/_pyinstaller/tests/pyinstaller-smoke.py +32 -0
  222. numpy/_pyinstaller/tests/test_pyinstaller.py +35 -0
  223. numpy/_pytesttester.py +201 -0
  224. numpy/_pytesttester.pyi +18 -0
  225. numpy/_typing/__init__.py +148 -0
  226. numpy/_typing/_add_docstring.py +153 -0
  227. numpy/_typing/_array_like.py +106 -0
  228. numpy/_typing/_char_codes.py +213 -0
  229. numpy/_typing/_dtype_like.py +114 -0
  230. numpy/_typing/_extended_precision.py +15 -0
  231. numpy/_typing/_nbit.py +19 -0
  232. numpy/_typing/_nbit_base.py +94 -0
  233. numpy/_typing/_nbit_base.pyi +40 -0
  234. numpy/_typing/_nested_sequence.py +79 -0
  235. numpy/_typing/_scalars.py +20 -0
  236. numpy/_typing/_shape.py +8 -0
  237. numpy/_typing/_ufunc.py +7 -0
  238. numpy/_typing/_ufunc.pyi +941 -0
  239. numpy/_utils/__init__.py +95 -0
  240. numpy/_utils/__init__.pyi +30 -0
  241. numpy/_utils/_convertions.py +18 -0
  242. numpy/_utils/_convertions.pyi +4 -0
  243. numpy/_utils/_inspect.py +192 -0
  244. numpy/_utils/_inspect.pyi +71 -0
  245. numpy/_utils/_pep440.py +486 -0
  246. numpy/_utils/_pep440.pyi +121 -0
  247. numpy/char/__init__.py +2 -0
  248. numpy/char/__init__.pyi +111 -0
  249. numpy/conftest.py +258 -0
  250. numpy/core/__init__.py +33 -0
  251. numpy/core/__init__.pyi +0 -0
  252. numpy/core/_dtype.py +10 -0
  253. numpy/core/_dtype.pyi +0 -0
  254. numpy/core/_dtype_ctypes.py +10 -0
  255. numpy/core/_dtype_ctypes.pyi +0 -0
  256. numpy/core/_internal.py +27 -0
  257. numpy/core/_multiarray_umath.py +57 -0
  258. numpy/core/_utils.py +21 -0
  259. numpy/core/arrayprint.py +10 -0
  260. numpy/core/defchararray.py +10 -0
  261. numpy/core/einsumfunc.py +10 -0
  262. numpy/core/fromnumeric.py +10 -0
  263. numpy/core/function_base.py +10 -0
  264. numpy/core/getlimits.py +10 -0
  265. numpy/core/multiarray.py +25 -0
  266. numpy/core/numeric.py +12 -0
  267. numpy/core/numerictypes.py +10 -0
  268. numpy/core/overrides.py +10 -0
  269. numpy/core/overrides.pyi +7 -0
  270. numpy/core/records.py +10 -0
  271. numpy/core/shape_base.py +10 -0
  272. numpy/core/umath.py +10 -0
  273. numpy/ctypeslib/__init__.py +13 -0
  274. numpy/ctypeslib/__init__.pyi +33 -0
  275. numpy/ctypeslib/_ctypeslib.py +603 -0
  276. numpy/ctypeslib/_ctypeslib.pyi +245 -0
  277. numpy/doc/ufuncs.py +138 -0
  278. numpy/dtypes.py +41 -0
  279. numpy/dtypes.pyi +631 -0
  280. numpy/exceptions.py +247 -0
  281. numpy/exceptions.pyi +27 -0
  282. numpy/f2py/__init__.py +86 -0
  283. numpy/f2py/__init__.pyi +6 -0
  284. numpy/f2py/__main__.py +5 -0
  285. numpy/f2py/__version__.py +1 -0
  286. numpy/f2py/__version__.pyi +1 -0
  287. numpy/f2py/_backends/__init__.py +9 -0
  288. numpy/f2py/_backends/__init__.pyi +5 -0
  289. numpy/f2py/_backends/_backend.py +44 -0
  290. numpy/f2py/_backends/_backend.pyi +46 -0
  291. numpy/f2py/_backends/_distutils.py +76 -0
  292. numpy/f2py/_backends/_distutils.pyi +13 -0
  293. numpy/f2py/_backends/_meson.py +231 -0
  294. numpy/f2py/_backends/_meson.pyi +63 -0
  295. numpy/f2py/_backends/meson.build.template +55 -0
  296. numpy/f2py/_isocbind.py +62 -0
  297. numpy/f2py/_isocbind.pyi +13 -0
  298. numpy/f2py/_src_pyf.py +247 -0
  299. numpy/f2py/_src_pyf.pyi +29 -0
  300. numpy/f2py/auxfuncs.py +1004 -0
  301. numpy/f2py/auxfuncs.pyi +264 -0
  302. numpy/f2py/capi_maps.py +811 -0
  303. numpy/f2py/capi_maps.pyi +33 -0
  304. numpy/f2py/cb_rules.py +665 -0
  305. numpy/f2py/cb_rules.pyi +17 -0
  306. numpy/f2py/cfuncs.py +1563 -0
  307. numpy/f2py/cfuncs.pyi +31 -0
  308. numpy/f2py/common_rules.py +143 -0
  309. numpy/f2py/common_rules.pyi +9 -0
  310. numpy/f2py/crackfortran.py +3725 -0
  311. numpy/f2py/crackfortran.pyi +258 -0
  312. numpy/f2py/diagnose.py +149 -0
  313. numpy/f2py/diagnose.pyi +1 -0
  314. numpy/f2py/f2py2e.py +786 -0
  315. numpy/f2py/f2py2e.pyi +76 -0
  316. numpy/f2py/f90mod_rules.py +269 -0
  317. numpy/f2py/f90mod_rules.pyi +16 -0
  318. numpy/f2py/func2subr.py +329 -0
  319. numpy/f2py/func2subr.pyi +7 -0
  320. numpy/f2py/rules.py +1629 -0
  321. numpy/f2py/rules.pyi +43 -0
  322. numpy/f2py/setup.cfg +3 -0
  323. numpy/f2py/src/fortranobject.c +1436 -0
  324. numpy/f2py/src/fortranobject.h +173 -0
  325. numpy/f2py/symbolic.py +1516 -0
  326. numpy/f2py/symbolic.pyi +221 -0
  327. numpy/f2py/tests/__init__.py +16 -0
  328. numpy/f2py/tests/src/abstract_interface/foo.f90 +34 -0
  329. numpy/f2py/tests/src/abstract_interface/gh18403_mod.f90 +6 -0
  330. numpy/f2py/tests/src/array_from_pyobj/wrapmodule.c +235 -0
  331. numpy/f2py/tests/src/assumed_shape/.f2py_f2cmap +1 -0
  332. numpy/f2py/tests/src/assumed_shape/foo_free.f90 +34 -0
  333. numpy/f2py/tests/src/assumed_shape/foo_mod.f90 +41 -0
  334. numpy/f2py/tests/src/assumed_shape/foo_use.f90 +19 -0
  335. numpy/f2py/tests/src/assumed_shape/precision.f90 +4 -0
  336. numpy/f2py/tests/src/block_docstring/foo.f +6 -0
  337. numpy/f2py/tests/src/callback/foo.f +62 -0
  338. numpy/f2py/tests/src/callback/gh17797.f90 +7 -0
  339. numpy/f2py/tests/src/callback/gh18335.f90 +17 -0
  340. numpy/f2py/tests/src/callback/gh25211.f +10 -0
  341. numpy/f2py/tests/src/callback/gh25211.pyf +18 -0
  342. numpy/f2py/tests/src/callback/gh26681.f90 +18 -0
  343. numpy/f2py/tests/src/cli/gh_22819.pyf +6 -0
  344. numpy/f2py/tests/src/cli/hi77.f +3 -0
  345. numpy/f2py/tests/src/cli/hiworld.f90 +3 -0
  346. numpy/f2py/tests/src/common/block.f +11 -0
  347. numpy/f2py/tests/src/common/gh19161.f90 +10 -0
  348. numpy/f2py/tests/src/crackfortran/accesstype.f90 +13 -0
  349. numpy/f2py/tests/src/crackfortran/common_with_division.f +17 -0
  350. numpy/f2py/tests/src/crackfortran/data_common.f +8 -0
  351. numpy/f2py/tests/src/crackfortran/data_multiplier.f +5 -0
  352. numpy/f2py/tests/src/crackfortran/data_stmts.f90 +20 -0
  353. numpy/f2py/tests/src/crackfortran/data_with_comments.f +8 -0
  354. numpy/f2py/tests/src/crackfortran/foo_deps.f90 +6 -0
  355. numpy/f2py/tests/src/crackfortran/gh15035.f +16 -0
  356. numpy/f2py/tests/src/crackfortran/gh17859.f +12 -0
  357. numpy/f2py/tests/src/crackfortran/gh22648.pyf +7 -0
  358. numpy/f2py/tests/src/crackfortran/gh23533.f +5 -0
  359. numpy/f2py/tests/src/crackfortran/gh23598.f90 +4 -0
  360. numpy/f2py/tests/src/crackfortran/gh23598Warn.f90 +11 -0
  361. numpy/f2py/tests/src/crackfortran/gh23879.f90 +20 -0
  362. numpy/f2py/tests/src/crackfortran/gh27697.f90 +12 -0
  363. numpy/f2py/tests/src/crackfortran/gh2848.f90 +13 -0
  364. numpy/f2py/tests/src/crackfortran/operators.f90 +49 -0
  365. numpy/f2py/tests/src/crackfortran/privatemod.f90 +11 -0
  366. numpy/f2py/tests/src/crackfortran/publicmod.f90 +10 -0
  367. numpy/f2py/tests/src/crackfortran/pubprivmod.f90 +10 -0
  368. numpy/f2py/tests/src/crackfortran/unicode_comment.f90 +4 -0
  369. numpy/f2py/tests/src/f2cmap/.f2py_f2cmap +1 -0
  370. numpy/f2py/tests/src/f2cmap/isoFortranEnvMap.f90 +9 -0
  371. numpy/f2py/tests/src/isocintrin/isoCtests.f90 +34 -0
  372. numpy/f2py/tests/src/kind/foo.f90 +20 -0
  373. numpy/f2py/tests/src/mixed/foo.f +5 -0
  374. numpy/f2py/tests/src/mixed/foo_fixed.f90 +8 -0
  375. numpy/f2py/tests/src/mixed/foo_free.f90 +8 -0
  376. numpy/f2py/tests/src/modules/gh25337/data.f90 +8 -0
  377. numpy/f2py/tests/src/modules/gh25337/use_data.f90 +6 -0
  378. numpy/f2py/tests/src/modules/gh26920/two_mods_with_no_public_entities.f90 +21 -0
  379. numpy/f2py/tests/src/modules/gh26920/two_mods_with_one_public_routine.f90 +21 -0
  380. numpy/f2py/tests/src/modules/module_data_docstring.f90 +12 -0
  381. numpy/f2py/tests/src/modules/use_modules.f90 +20 -0
  382. numpy/f2py/tests/src/negative_bounds/issue_20853.f90 +7 -0
  383. numpy/f2py/tests/src/parameter/constant_array.f90 +45 -0
  384. numpy/f2py/tests/src/parameter/constant_both.f90 +57 -0
  385. numpy/f2py/tests/src/parameter/constant_compound.f90 +15 -0
  386. numpy/f2py/tests/src/parameter/constant_integer.f90 +22 -0
  387. numpy/f2py/tests/src/parameter/constant_non_compound.f90 +23 -0
  388. numpy/f2py/tests/src/parameter/constant_real.f90 +23 -0
  389. numpy/f2py/tests/src/quoted_character/foo.f +14 -0
  390. numpy/f2py/tests/src/regression/AB.inc +1 -0
  391. numpy/f2py/tests/src/regression/assignOnlyModule.f90 +25 -0
  392. numpy/f2py/tests/src/regression/datonly.f90 +17 -0
  393. numpy/f2py/tests/src/regression/f77comments.f +26 -0
  394. numpy/f2py/tests/src/regression/f77fixedform.f95 +5 -0
  395. numpy/f2py/tests/src/regression/f90continuation.f90 +9 -0
  396. numpy/f2py/tests/src/regression/incfile.f90 +5 -0
  397. numpy/f2py/tests/src/regression/inout.f90 +9 -0
  398. numpy/f2py/tests/src/regression/lower_f2py_fortran.f90 +5 -0
  399. numpy/f2py/tests/src/regression/mod_derived_types.f90 +23 -0
  400. numpy/f2py/tests/src/return_character/foo77.f +45 -0
  401. numpy/f2py/tests/src/return_character/foo90.f90 +48 -0
  402. numpy/f2py/tests/src/return_complex/foo77.f +45 -0
  403. numpy/f2py/tests/src/return_complex/foo90.f90 +48 -0
  404. numpy/f2py/tests/src/return_integer/foo77.f +56 -0
  405. numpy/f2py/tests/src/return_integer/foo90.f90 +59 -0
  406. numpy/f2py/tests/src/return_logical/foo77.f +56 -0
  407. numpy/f2py/tests/src/return_logical/foo90.f90 +59 -0
  408. numpy/f2py/tests/src/return_real/foo77.f +45 -0
  409. numpy/f2py/tests/src/return_real/foo90.f90 +48 -0
  410. numpy/f2py/tests/src/routines/funcfortranname.f +5 -0
  411. numpy/f2py/tests/src/routines/funcfortranname.pyf +11 -0
  412. numpy/f2py/tests/src/routines/subrout.f +4 -0
  413. numpy/f2py/tests/src/routines/subrout.pyf +10 -0
  414. numpy/f2py/tests/src/size/foo.f90 +44 -0
  415. numpy/f2py/tests/src/string/char.f90 +29 -0
  416. numpy/f2py/tests/src/string/fixed_string.f90 +34 -0
  417. numpy/f2py/tests/src/string/gh24008.f +8 -0
  418. numpy/f2py/tests/src/string/gh24662.f90 +7 -0
  419. numpy/f2py/tests/src/string/gh25286.f90 +14 -0
  420. numpy/f2py/tests/src/string/gh25286.pyf +12 -0
  421. numpy/f2py/tests/src/string/gh25286_bc.pyf +12 -0
  422. numpy/f2py/tests/src/string/scalar_string.f90 +9 -0
  423. numpy/f2py/tests/src/string/string.f +12 -0
  424. numpy/f2py/tests/src/value_attrspec/gh21665.f90 +9 -0
  425. numpy/f2py/tests/test_abstract_interface.py +26 -0
  426. numpy/f2py/tests/test_array_from_pyobj.py +678 -0
  427. numpy/f2py/tests/test_assumed_shape.py +50 -0
  428. numpy/f2py/tests/test_block_docstring.py +20 -0
  429. numpy/f2py/tests/test_callback.py +263 -0
  430. numpy/f2py/tests/test_character.py +641 -0
  431. numpy/f2py/tests/test_common.py +23 -0
  432. numpy/f2py/tests/test_crackfortran.py +421 -0
  433. numpy/f2py/tests/test_data.py +71 -0
  434. numpy/f2py/tests/test_docs.py +64 -0
  435. numpy/f2py/tests/test_f2cmap.py +17 -0
  436. numpy/f2py/tests/test_f2py2e.py +964 -0
  437. numpy/f2py/tests/test_isoc.py +56 -0
  438. numpy/f2py/tests/test_kind.py +53 -0
  439. numpy/f2py/tests/test_mixed.py +35 -0
  440. numpy/f2py/tests/test_modules.py +83 -0
  441. numpy/f2py/tests/test_parameter.py +129 -0
  442. numpy/f2py/tests/test_pyf_src.py +43 -0
  443. numpy/f2py/tests/test_quoted_character.py +18 -0
  444. numpy/f2py/tests/test_regression.py +187 -0
  445. numpy/f2py/tests/test_return_character.py +48 -0
  446. numpy/f2py/tests/test_return_complex.py +67 -0
  447. numpy/f2py/tests/test_return_integer.py +55 -0
  448. numpy/f2py/tests/test_return_logical.py +65 -0
  449. numpy/f2py/tests/test_return_real.py +109 -0
  450. numpy/f2py/tests/test_routines.py +29 -0
  451. numpy/f2py/tests/test_semicolon_split.py +75 -0
  452. numpy/f2py/tests/test_size.py +45 -0
  453. numpy/f2py/tests/test_string.py +100 -0
  454. numpy/f2py/tests/test_symbolic.py +495 -0
  455. numpy/f2py/tests/test_value_attrspec.py +15 -0
  456. numpy/f2py/tests/util.py +442 -0
  457. numpy/f2py/use_rules.py +99 -0
  458. numpy/f2py/use_rules.pyi +9 -0
  459. numpy/fft/__init__.py +215 -0
  460. numpy/fft/__init__.pyi +43 -0
  461. numpy/fft/_helper.py +235 -0
  462. numpy/fft/_helper.pyi +45 -0
  463. numpy/fft/_pocketfft.py +1693 -0
  464. numpy/fft/_pocketfft.pyi +138 -0
  465. numpy/fft/_pocketfft_umath.cpython-313-darwin.so +0 -0
  466. numpy/fft/helper.py +17 -0
  467. numpy/fft/helper.pyi +22 -0
  468. numpy/fft/tests/__init__.py +0 -0
  469. numpy/fft/tests/test_helper.py +167 -0
  470. numpy/fft/tests/test_pocketfft.py +589 -0
  471. numpy/lib/__init__.py +97 -0
  472. numpy/lib/__init__.pyi +44 -0
  473. numpy/lib/_array_utils_impl.py +62 -0
  474. numpy/lib/_array_utils_impl.pyi +26 -0
  475. numpy/lib/_arraypad_impl.py +890 -0
  476. numpy/lib/_arraypad_impl.pyi +89 -0
  477. numpy/lib/_arraysetops_impl.py +1260 -0
  478. numpy/lib/_arraysetops_impl.pyi +468 -0
  479. numpy/lib/_arrayterator_impl.py +224 -0
  480. numpy/lib/_arrayterator_impl.pyi +46 -0
  481. numpy/lib/_datasource.py +700 -0
  482. numpy/lib/_datasource.pyi +31 -0
  483. numpy/lib/_format_impl.py +1036 -0
  484. numpy/lib/_format_impl.pyi +26 -0
  485. numpy/lib/_function_base_impl.py +5844 -0
  486. numpy/lib/_function_base_impl.pyi +1164 -0
  487. numpy/lib/_histograms_impl.py +1085 -0
  488. numpy/lib/_histograms_impl.pyi +50 -0
  489. numpy/lib/_index_tricks_impl.py +1067 -0
  490. numpy/lib/_index_tricks_impl.pyi +208 -0
  491. numpy/lib/_iotools.py +900 -0
  492. numpy/lib/_iotools.pyi +114 -0
  493. numpy/lib/_nanfunctions_impl.py +2024 -0
  494. numpy/lib/_nanfunctions_impl.pyi +52 -0
  495. numpy/lib/_npyio_impl.py +2596 -0
  496. numpy/lib/_npyio_impl.pyi +301 -0
  497. numpy/lib/_polynomial_impl.py +1465 -0
  498. numpy/lib/_polynomial_impl.pyi +318 -0
  499. numpy/lib/_scimath_impl.py +642 -0
  500. numpy/lib/_scimath_impl.pyi +93 -0
  501. numpy/lib/_shape_base_impl.py +1301 -0
  502. numpy/lib/_shape_base_impl.pyi +235 -0
  503. numpy/lib/_stride_tricks_impl.py +549 -0
  504. numpy/lib/_stride_tricks_impl.pyi +74 -0
  505. numpy/lib/_twodim_base_impl.py +1201 -0
  506. numpy/lib/_twodim_base_impl.pyi +438 -0
  507. numpy/lib/_type_check_impl.py +699 -0
  508. numpy/lib/_type_check_impl.pyi +350 -0
  509. numpy/lib/_ufunclike_impl.py +207 -0
  510. numpy/lib/_ufunclike_impl.pyi +67 -0
  511. numpy/lib/_user_array_impl.py +299 -0
  512. numpy/lib/_user_array_impl.pyi +225 -0
  513. numpy/lib/_utils_impl.py +784 -0
  514. numpy/lib/_utils_impl.pyi +10 -0
  515. numpy/lib/_version.py +154 -0
  516. numpy/lib/_version.pyi +17 -0
  517. numpy/lib/array_utils.py +7 -0
  518. numpy/lib/array_utils.pyi +12 -0
  519. numpy/lib/format.py +24 -0
  520. numpy/lib/format.pyi +66 -0
  521. numpy/lib/introspect.py +95 -0
  522. numpy/lib/introspect.pyi +3 -0
  523. numpy/lib/mixins.py +180 -0
  524. numpy/lib/mixins.pyi +77 -0
  525. numpy/lib/npyio.py +1 -0
  526. numpy/lib/npyio.pyi +9 -0
  527. numpy/lib/recfunctions.py +1681 -0
  528. numpy/lib/recfunctions.pyi +435 -0
  529. numpy/lib/scimath.py +13 -0
  530. numpy/lib/scimath.pyi +30 -0
  531. numpy/lib/stride_tricks.py +1 -0
  532. numpy/lib/stride_tricks.pyi +6 -0
  533. numpy/lib/tests/__init__.py +0 -0
  534. numpy/lib/tests/data/py2-np0-objarr.npy +0 -0
  535. numpy/lib/tests/data/py2-objarr.npy +0 -0
  536. numpy/lib/tests/data/py2-objarr.npz +0 -0
  537. numpy/lib/tests/data/py3-objarr.npy +0 -0
  538. numpy/lib/tests/data/py3-objarr.npz +0 -0
  539. numpy/lib/tests/data/python3.npy +0 -0
  540. numpy/lib/tests/data/win64python2.npy +0 -0
  541. numpy/lib/tests/test__datasource.py +352 -0
  542. numpy/lib/tests/test__iotools.py +360 -0
  543. numpy/lib/tests/test__version.py +64 -0
  544. numpy/lib/tests/test_array_utils.py +32 -0
  545. numpy/lib/tests/test_arraypad.py +1415 -0
  546. numpy/lib/tests/test_arraysetops.py +1074 -0
  547. numpy/lib/tests/test_arrayterator.py +46 -0
  548. numpy/lib/tests/test_format.py +1054 -0
  549. numpy/lib/tests/test_function_base.py +4573 -0
  550. numpy/lib/tests/test_histograms.py +855 -0
  551. numpy/lib/tests/test_index_tricks.py +573 -0
  552. numpy/lib/tests/test_io.py +2848 -0
  553. numpy/lib/tests/test_loadtxt.py +1101 -0
  554. numpy/lib/tests/test_mixins.py +215 -0
  555. numpy/lib/tests/test_nanfunctions.py +1438 -0
  556. numpy/lib/tests/test_packbits.py +376 -0
  557. numpy/lib/tests/test_polynomial.py +320 -0
  558. numpy/lib/tests/test_recfunctions.py +1052 -0
  559. numpy/lib/tests/test_regression.py +231 -0
  560. numpy/lib/tests/test_shape_base.py +813 -0
  561. numpy/lib/tests/test_stride_tricks.py +656 -0
  562. numpy/lib/tests/test_twodim_base.py +559 -0
  563. numpy/lib/tests/test_type_check.py +473 -0
  564. numpy/lib/tests/test_ufunclike.py +97 -0
  565. numpy/lib/tests/test_utils.py +80 -0
  566. numpy/lib/user_array.py +1 -0
  567. numpy/lib/user_array.pyi +1 -0
  568. numpy/linalg/__init__.py +98 -0
  569. numpy/linalg/__init__.pyi +73 -0
  570. numpy/linalg/_linalg.py +3682 -0
  571. numpy/linalg/_linalg.pyi +475 -0
  572. numpy/linalg/_umath_linalg.cpython-313-darwin.so +0 -0
  573. numpy/linalg/_umath_linalg.pyi +61 -0
  574. numpy/linalg/lapack_lite.cpython-313-darwin.so +0 -0
  575. numpy/linalg/lapack_lite.pyi +141 -0
  576. numpy/linalg/linalg.py +17 -0
  577. numpy/linalg/linalg.pyi +69 -0
  578. numpy/linalg/tests/__init__.py +0 -0
  579. numpy/linalg/tests/test_deprecations.py +20 -0
  580. numpy/linalg/tests/test_linalg.py +2443 -0
  581. numpy/linalg/tests/test_regression.py +181 -0
  582. numpy/ma/API_CHANGES.txt +135 -0
  583. numpy/ma/LICENSE +24 -0
  584. numpy/ma/README.rst +236 -0
  585. numpy/ma/__init__.py +53 -0
  586. numpy/ma/__init__.pyi +458 -0
  587. numpy/ma/core.py +8933 -0
  588. numpy/ma/core.pyi +1462 -0
  589. numpy/ma/extras.py +2344 -0
  590. numpy/ma/extras.pyi +138 -0
  591. numpy/ma/mrecords.py +773 -0
  592. numpy/ma/mrecords.pyi +96 -0
  593. numpy/ma/tests/__init__.py +0 -0
  594. numpy/ma/tests/test_arrayobject.py +40 -0
  595. numpy/ma/tests/test_core.py +5886 -0
  596. numpy/ma/tests/test_deprecations.py +87 -0
  597. numpy/ma/tests/test_extras.py +1998 -0
  598. numpy/ma/tests/test_mrecords.py +497 -0
  599. numpy/ma/tests/test_old_ma.py +942 -0
  600. numpy/ma/tests/test_regression.py +100 -0
  601. numpy/ma/tests/test_subclassing.py +469 -0
  602. numpy/ma/testutils.py +294 -0
  603. numpy/matlib.py +380 -0
  604. numpy/matlib.pyi +582 -0
  605. numpy/matrixlib/__init__.py +12 -0
  606. numpy/matrixlib/__init__.pyi +5 -0
  607. numpy/matrixlib/defmatrix.py +1119 -0
  608. numpy/matrixlib/defmatrix.pyi +17 -0
  609. numpy/matrixlib/tests/__init__.py +0 -0
  610. numpy/matrixlib/tests/test_defmatrix.py +455 -0
  611. numpy/matrixlib/tests/test_interaction.py +360 -0
  612. numpy/matrixlib/tests/test_masked_matrix.py +240 -0
  613. numpy/matrixlib/tests/test_matrix_linalg.py +105 -0
  614. numpy/matrixlib/tests/test_multiarray.py +17 -0
  615. numpy/matrixlib/tests/test_numeric.py +18 -0
  616. numpy/matrixlib/tests/test_regression.py +31 -0
  617. numpy/polynomial/__init__.py +187 -0
  618. numpy/polynomial/__init__.pyi +25 -0
  619. numpy/polynomial/_polybase.py +1191 -0
  620. numpy/polynomial/_polybase.pyi +285 -0
  621. numpy/polynomial/_polytypes.pyi +892 -0
  622. numpy/polynomial/chebyshev.py +2003 -0
  623. numpy/polynomial/chebyshev.pyi +181 -0
  624. numpy/polynomial/hermite.py +1740 -0
  625. numpy/polynomial/hermite.pyi +107 -0
  626. numpy/polynomial/hermite_e.py +1642 -0
  627. numpy/polynomial/hermite_e.pyi +107 -0
  628. numpy/polynomial/laguerre.py +1675 -0
  629. numpy/polynomial/laguerre.pyi +100 -0
  630. numpy/polynomial/legendre.py +1605 -0
  631. numpy/polynomial/legendre.pyi +100 -0
  632. numpy/polynomial/polynomial.py +1616 -0
  633. numpy/polynomial/polynomial.pyi +89 -0
  634. numpy/polynomial/polyutils.py +759 -0
  635. numpy/polynomial/polyutils.pyi +423 -0
  636. numpy/polynomial/tests/__init__.py +0 -0
  637. numpy/polynomial/tests/test_chebyshev.py +623 -0
  638. numpy/polynomial/tests/test_classes.py +618 -0
  639. numpy/polynomial/tests/test_hermite.py +558 -0
  640. numpy/polynomial/tests/test_hermite_e.py +559 -0
  641. numpy/polynomial/tests/test_laguerre.py +540 -0
  642. numpy/polynomial/tests/test_legendre.py +571 -0
  643. numpy/polynomial/tests/test_polynomial.py +669 -0
  644. numpy/polynomial/tests/test_polyutils.py +128 -0
  645. numpy/polynomial/tests/test_printing.py +555 -0
  646. numpy/polynomial/tests/test_symbol.py +217 -0
  647. numpy/py.typed +0 -0
  648. numpy/random/LICENSE.md +71 -0
  649. numpy/random/__init__.pxd +14 -0
  650. numpy/random/__init__.py +213 -0
  651. numpy/random/__init__.pyi +124 -0
  652. numpy/random/_bounded_integers.cpython-313-darwin.so +0 -0
  653. numpy/random/_bounded_integers.pxd +29 -0
  654. numpy/random/_bounded_integers.pyi +1 -0
  655. numpy/random/_common.cpython-313-darwin.so +0 -0
  656. numpy/random/_common.pxd +107 -0
  657. numpy/random/_common.pyi +16 -0
  658. numpy/random/_examples/cffi/extending.py +44 -0
  659. numpy/random/_examples/cffi/parse.py +53 -0
  660. numpy/random/_examples/cython/extending.pyx +77 -0
  661. numpy/random/_examples/cython/extending_distributions.pyx +118 -0
  662. numpy/random/_examples/cython/meson.build +53 -0
  663. numpy/random/_examples/numba/extending.py +86 -0
  664. numpy/random/_examples/numba/extending_distributions.py +67 -0
  665. numpy/random/_generator.cpython-313-darwin.so +0 -0
  666. numpy/random/_generator.pyi +861 -0
  667. numpy/random/_mt19937.cpython-313-darwin.so +0 -0
  668. numpy/random/_mt19937.pyi +25 -0
  669. numpy/random/_pcg64.cpython-313-darwin.so +0 -0
  670. numpy/random/_pcg64.pyi +44 -0
  671. numpy/random/_philox.cpython-313-darwin.so +0 -0
  672. numpy/random/_philox.pyi +39 -0
  673. numpy/random/_pickle.py +88 -0
  674. numpy/random/_pickle.pyi +43 -0
  675. numpy/random/_sfc64.cpython-313-darwin.so +0 -0
  676. numpy/random/_sfc64.pyi +28 -0
  677. numpy/random/bit_generator.cpython-313-darwin.so +0 -0
  678. numpy/random/bit_generator.pxd +35 -0
  679. numpy/random/bit_generator.pyi +124 -0
  680. numpy/random/c_distributions.pxd +119 -0
  681. numpy/random/lib/libnpyrandom.a +0 -0
  682. numpy/random/mtrand.cpython-313-darwin.so +0 -0
  683. numpy/random/mtrand.pyi +703 -0
  684. numpy/random/tests/__init__.py +0 -0
  685. numpy/random/tests/data/__init__.py +0 -0
  686. numpy/random/tests/data/generator_pcg64_np121.pkl.gz +0 -0
  687. numpy/random/tests/data/generator_pcg64_np126.pkl.gz +0 -0
  688. numpy/random/tests/data/mt19937-testset-1.csv +1001 -0
  689. numpy/random/tests/data/mt19937-testset-2.csv +1001 -0
  690. numpy/random/tests/data/pcg64-testset-1.csv +1001 -0
  691. numpy/random/tests/data/pcg64-testset-2.csv +1001 -0
  692. numpy/random/tests/data/pcg64dxsm-testset-1.csv +1001 -0
  693. numpy/random/tests/data/pcg64dxsm-testset-2.csv +1001 -0
  694. numpy/random/tests/data/philox-testset-1.csv +1001 -0
  695. numpy/random/tests/data/philox-testset-2.csv +1001 -0
  696. numpy/random/tests/data/sfc64-testset-1.csv +1001 -0
  697. numpy/random/tests/data/sfc64-testset-2.csv +1001 -0
  698. numpy/random/tests/data/sfc64_np126.pkl.gz +0 -0
  699. numpy/random/tests/test_direct.py +592 -0
  700. numpy/random/tests/test_extending.py +127 -0
  701. numpy/random/tests/test_generator_mt19937.py +2809 -0
  702. numpy/random/tests/test_generator_mt19937_regressions.py +207 -0
  703. numpy/random/tests/test_random.py +1757 -0
  704. numpy/random/tests/test_randomstate.py +2130 -0
  705. numpy/random/tests/test_randomstate_regression.py +217 -0
  706. numpy/random/tests/test_regression.py +152 -0
  707. numpy/random/tests/test_seed_sequence.py +79 -0
  708. numpy/random/tests/test_smoke.py +819 -0
  709. numpy/rec/__init__.py +2 -0
  710. numpy/rec/__init__.pyi +23 -0
  711. numpy/strings/__init__.py +2 -0
  712. numpy/strings/__init__.pyi +97 -0
  713. numpy/testing/__init__.py +22 -0
  714. numpy/testing/__init__.pyi +102 -0
  715. numpy/testing/_private/__init__.py +0 -0
  716. numpy/testing/_private/__init__.pyi +0 -0
  717. numpy/testing/_private/extbuild.py +250 -0
  718. numpy/testing/_private/extbuild.pyi +25 -0
  719. numpy/testing/_private/utils.py +2752 -0
  720. numpy/testing/_private/utils.pyi +499 -0
  721. numpy/testing/overrides.py +84 -0
  722. numpy/testing/overrides.pyi +11 -0
  723. numpy/testing/print_coercion_tables.py +207 -0
  724. numpy/testing/print_coercion_tables.pyi +27 -0
  725. numpy/testing/tests/__init__.py +0 -0
  726. numpy/testing/tests/test_utils.py +1917 -0
  727. numpy/tests/__init__.py +0 -0
  728. numpy/tests/test__all__.py +10 -0
  729. numpy/tests/test_configtool.py +48 -0
  730. numpy/tests/test_ctypeslib.py +377 -0
  731. numpy/tests/test_lazyloading.py +38 -0
  732. numpy/tests/test_matlib.py +59 -0
  733. numpy/tests/test_numpy_config.py +46 -0
  734. numpy/tests/test_numpy_version.py +54 -0
  735. numpy/tests/test_public_api.py +806 -0
  736. numpy/tests/test_reloading.py +74 -0
  737. numpy/tests/test_scripts.py +49 -0
  738. numpy/tests/test_warnings.py +78 -0
  739. numpy/typing/__init__.py +201 -0
  740. numpy/typing/mypy_plugin.py +195 -0
  741. numpy/typing/tests/__init__.py +0 -0
  742. numpy/typing/tests/data/fail/arithmetic.pyi +126 -0
  743. numpy/typing/tests/data/fail/array_constructors.pyi +34 -0
  744. numpy/typing/tests/data/fail/array_like.pyi +15 -0
  745. numpy/typing/tests/data/fail/array_pad.pyi +6 -0
  746. numpy/typing/tests/data/fail/arrayprint.pyi +16 -0
  747. numpy/typing/tests/data/fail/arrayterator.pyi +14 -0
  748. numpy/typing/tests/data/fail/bitwise_ops.pyi +17 -0
  749. numpy/typing/tests/data/fail/char.pyi +65 -0
  750. numpy/typing/tests/data/fail/chararray.pyi +62 -0
  751. numpy/typing/tests/data/fail/comparisons.pyi +27 -0
  752. numpy/typing/tests/data/fail/constants.pyi +3 -0
  753. numpy/typing/tests/data/fail/datasource.pyi +15 -0
  754. numpy/typing/tests/data/fail/dtype.pyi +17 -0
  755. numpy/typing/tests/data/fail/einsumfunc.pyi +12 -0
  756. numpy/typing/tests/data/fail/flatiter.pyi +20 -0
  757. numpy/typing/tests/data/fail/fromnumeric.pyi +148 -0
  758. numpy/typing/tests/data/fail/histograms.pyi +12 -0
  759. numpy/typing/tests/data/fail/index_tricks.pyi +14 -0
  760. numpy/typing/tests/data/fail/lib_function_base.pyi +62 -0
  761. numpy/typing/tests/data/fail/lib_polynomial.pyi +29 -0
  762. numpy/typing/tests/data/fail/lib_utils.pyi +3 -0
  763. numpy/typing/tests/data/fail/lib_version.pyi +6 -0
  764. numpy/typing/tests/data/fail/linalg.pyi +48 -0
  765. numpy/typing/tests/data/fail/ma.pyi +143 -0
  766. numpy/typing/tests/data/fail/memmap.pyi +5 -0
  767. numpy/typing/tests/data/fail/modules.pyi +17 -0
  768. numpy/typing/tests/data/fail/multiarray.pyi +52 -0
  769. numpy/typing/tests/data/fail/ndarray.pyi +11 -0
  770. numpy/typing/tests/data/fail/ndarray_misc.pyi +36 -0
  771. numpy/typing/tests/data/fail/nditer.pyi +8 -0
  772. numpy/typing/tests/data/fail/nested_sequence.pyi +16 -0
  773. numpy/typing/tests/data/fail/npyio.pyi +24 -0
  774. numpy/typing/tests/data/fail/numerictypes.pyi +5 -0
  775. numpy/typing/tests/data/fail/random.pyi +62 -0
  776. numpy/typing/tests/data/fail/rec.pyi +17 -0
  777. numpy/typing/tests/data/fail/scalars.pyi +87 -0
  778. numpy/typing/tests/data/fail/shape.pyi +6 -0
  779. numpy/typing/tests/data/fail/shape_base.pyi +8 -0
  780. numpy/typing/tests/data/fail/stride_tricks.pyi +9 -0
  781. numpy/typing/tests/data/fail/strings.pyi +52 -0
  782. numpy/typing/tests/data/fail/testing.pyi +28 -0
  783. numpy/typing/tests/data/fail/twodim_base.pyi +32 -0
  784. numpy/typing/tests/data/fail/type_check.pyi +13 -0
  785. numpy/typing/tests/data/fail/ufunc_config.pyi +21 -0
  786. numpy/typing/tests/data/fail/ufunclike.pyi +21 -0
  787. numpy/typing/tests/data/fail/ufuncs.pyi +17 -0
  788. numpy/typing/tests/data/fail/warnings_and_errors.pyi +5 -0
  789. numpy/typing/tests/data/misc/extended_precision.pyi +9 -0
  790. numpy/typing/tests/data/mypy.ini +9 -0
  791. numpy/typing/tests/data/pass/arithmetic.py +612 -0
  792. numpy/typing/tests/data/pass/array_constructors.py +137 -0
  793. numpy/typing/tests/data/pass/array_like.py +43 -0
  794. numpy/typing/tests/data/pass/arrayprint.py +37 -0
  795. numpy/typing/tests/data/pass/arrayterator.py +27 -0
  796. numpy/typing/tests/data/pass/bitwise_ops.py +131 -0
  797. numpy/typing/tests/data/pass/comparisons.py +315 -0
  798. numpy/typing/tests/data/pass/dtype.py +57 -0
  799. numpy/typing/tests/data/pass/einsumfunc.py +36 -0
  800. numpy/typing/tests/data/pass/flatiter.py +19 -0
  801. numpy/typing/tests/data/pass/fromnumeric.py +272 -0
  802. numpy/typing/tests/data/pass/index_tricks.py +60 -0
  803. numpy/typing/tests/data/pass/lib_user_array.py +22 -0
  804. numpy/typing/tests/data/pass/lib_utils.py +19 -0
  805. numpy/typing/tests/data/pass/lib_version.py +18 -0
  806. numpy/typing/tests/data/pass/literal.py +51 -0
  807. numpy/typing/tests/data/pass/ma.py +174 -0
  808. numpy/typing/tests/data/pass/mod.py +149 -0
  809. numpy/typing/tests/data/pass/modules.py +45 -0
  810. numpy/typing/tests/data/pass/multiarray.py +76 -0
  811. numpy/typing/tests/data/pass/ndarray_conversion.py +87 -0
  812. numpy/typing/tests/data/pass/ndarray_misc.py +203 -0
  813. numpy/typing/tests/data/pass/ndarray_shape_manipulation.py +47 -0
  814. numpy/typing/tests/data/pass/nditer.py +4 -0
  815. numpy/typing/tests/data/pass/numeric.py +95 -0
  816. numpy/typing/tests/data/pass/numerictypes.py +17 -0
  817. numpy/typing/tests/data/pass/random.py +1497 -0
  818. numpy/typing/tests/data/pass/recfunctions.py +161 -0
  819. numpy/typing/tests/data/pass/scalars.py +248 -0
  820. numpy/typing/tests/data/pass/shape.py +19 -0
  821. numpy/typing/tests/data/pass/simple.py +168 -0
  822. numpy/typing/tests/data/pass/simple_py3.py +6 -0
  823. numpy/typing/tests/data/pass/ufunc_config.py +64 -0
  824. numpy/typing/tests/data/pass/ufunclike.py +47 -0
  825. numpy/typing/tests/data/pass/ufuncs.py +16 -0
  826. numpy/typing/tests/data/pass/warnings_and_errors.py +6 -0
  827. numpy/typing/tests/data/reveal/arithmetic.pyi +720 -0
  828. numpy/typing/tests/data/reveal/array_api_info.pyi +70 -0
  829. numpy/typing/tests/data/reveal/array_constructors.pyi +249 -0
  830. numpy/typing/tests/data/reveal/arraypad.pyi +22 -0
  831. numpy/typing/tests/data/reveal/arrayprint.pyi +25 -0
  832. numpy/typing/tests/data/reveal/arraysetops.pyi +74 -0
  833. numpy/typing/tests/data/reveal/arrayterator.pyi +27 -0
  834. numpy/typing/tests/data/reveal/bitwise_ops.pyi +167 -0
  835. numpy/typing/tests/data/reveal/char.pyi +224 -0
  836. numpy/typing/tests/data/reveal/chararray.pyi +137 -0
  837. numpy/typing/tests/data/reveal/comparisons.pyi +264 -0
  838. numpy/typing/tests/data/reveal/constants.pyi +14 -0
  839. numpy/typing/tests/data/reveal/ctypeslib.pyi +81 -0
  840. numpy/typing/tests/data/reveal/datasource.pyi +23 -0
  841. numpy/typing/tests/data/reveal/dtype.pyi +136 -0
  842. numpy/typing/tests/data/reveal/einsumfunc.pyi +39 -0
  843. numpy/typing/tests/data/reveal/emath.pyi +54 -0
  844. numpy/typing/tests/data/reveal/fft.pyi +37 -0
  845. numpy/typing/tests/data/reveal/flatiter.pyi +47 -0
  846. numpy/typing/tests/data/reveal/fromnumeric.pyi +347 -0
  847. numpy/typing/tests/data/reveal/getlimits.pyi +51 -0
  848. numpy/typing/tests/data/reveal/histograms.pyi +25 -0
  849. numpy/typing/tests/data/reveal/index_tricks.pyi +70 -0
  850. numpy/typing/tests/data/reveal/lib_function_base.pyi +213 -0
  851. numpy/typing/tests/data/reveal/lib_polynomial.pyi +144 -0
  852. numpy/typing/tests/data/reveal/lib_utils.pyi +17 -0
  853. numpy/typing/tests/data/reveal/lib_version.pyi +20 -0
  854. numpy/typing/tests/data/reveal/linalg.pyi +132 -0
  855. numpy/typing/tests/data/reveal/ma.pyi +369 -0
  856. numpy/typing/tests/data/reveal/matrix.pyi +73 -0
  857. numpy/typing/tests/data/reveal/memmap.pyi +19 -0
  858. numpy/typing/tests/data/reveal/mod.pyi +179 -0
  859. numpy/typing/tests/data/reveal/modules.pyi +51 -0
  860. numpy/typing/tests/data/reveal/multiarray.pyi +194 -0
  861. numpy/typing/tests/data/reveal/nbit_base_example.pyi +21 -0
  862. numpy/typing/tests/data/reveal/ndarray_assignability.pyi +77 -0
  863. numpy/typing/tests/data/reveal/ndarray_conversion.pyi +85 -0
  864. numpy/typing/tests/data/reveal/ndarray_misc.pyi +247 -0
  865. numpy/typing/tests/data/reveal/ndarray_shape_manipulation.pyi +39 -0
  866. numpy/typing/tests/data/reveal/nditer.pyi +49 -0
  867. numpy/typing/tests/data/reveal/nested_sequence.pyi +25 -0
  868. numpy/typing/tests/data/reveal/npyio.pyi +83 -0
  869. numpy/typing/tests/data/reveal/numeric.pyi +134 -0
  870. numpy/typing/tests/data/reveal/numerictypes.pyi +16 -0
  871. numpy/typing/tests/data/reveal/polynomial_polybase.pyi +220 -0
  872. numpy/typing/tests/data/reveal/polynomial_polyutils.pyi +219 -0
  873. numpy/typing/tests/data/reveal/polynomial_series.pyi +138 -0
  874. numpy/typing/tests/data/reveal/random.pyi +1546 -0
  875. numpy/typing/tests/data/reveal/rec.pyi +171 -0
  876. numpy/typing/tests/data/reveal/scalars.pyi +191 -0
  877. numpy/typing/tests/data/reveal/shape.pyi +13 -0
  878. numpy/typing/tests/data/reveal/shape_base.pyi +52 -0
  879. numpy/typing/tests/data/reveal/stride_tricks.pyi +27 -0
  880. numpy/typing/tests/data/reveal/strings.pyi +196 -0
  881. numpy/typing/tests/data/reveal/testing.pyi +198 -0
  882. numpy/typing/tests/data/reveal/twodim_base.pyi +145 -0
  883. numpy/typing/tests/data/reveal/type_check.pyi +67 -0
  884. numpy/typing/tests/data/reveal/ufunc_config.pyi +30 -0
  885. numpy/typing/tests/data/reveal/ufunclike.pyi +31 -0
  886. numpy/typing/tests/data/reveal/ufuncs.pyi +123 -0
  887. numpy/typing/tests/data/reveal/warnings_and_errors.pyi +11 -0
  888. numpy/typing/tests/test_isfile.py +32 -0
  889. numpy/typing/tests/test_runtime.py +102 -0
  890. numpy/typing/tests/test_typing.py +205 -0
  891. numpy/version.py +11 -0
  892. numpy/version.pyi +18 -0
  893. numpy-2.3.5.dist-info/LICENSE.txt +971 -0
  894. numpy-2.3.5.dist-info/METADATA +1093 -0
  895. numpy-2.3.5.dist-info/RECORD +897 -0
  896. numpy-2.3.5.dist-info/WHEEL +6 -0
  897. numpy-2.3.5.dist-info/entry_points.txt +13 -0
numpy/ma/core.pyi ADDED
@@ -0,0 +1,1462 @@
1
+ # pyright: reportIncompatibleMethodOverride=false
2
+ # ruff: noqa: ANN001, ANN002, ANN003, ANN201, ANN202 ANN204, ANN401
3
+
4
+ from collections.abc import Sequence
5
+ from typing import Any, Literal, Self, SupportsIndex, TypeAlias, overload
6
+
7
+ from _typeshed import Incomplete
8
+ from typing_extensions import TypeIs, TypeVar
9
+
10
+ import numpy as np
11
+ from numpy import (
12
+ _HasDTypeWithRealAndImag,
13
+ _ModeKind,
14
+ _OrderKACF,
15
+ _PartitionKind,
16
+ _SortKind,
17
+ amax,
18
+ amin,
19
+ bool_,
20
+ bytes_,
21
+ character,
22
+ complexfloating,
23
+ datetime64,
24
+ dtype,
25
+ dtypes,
26
+ expand_dims,
27
+ float64,
28
+ floating,
29
+ generic,
30
+ int_,
31
+ integer,
32
+ intp,
33
+ ndarray,
34
+ object_,
35
+ str_,
36
+ timedelta64,
37
+ )
38
+ from numpy._globals import _NoValueType
39
+ from numpy._typing import (
40
+ ArrayLike,
41
+ NDArray,
42
+ _AnyShape,
43
+ _ArrayLike,
44
+ _ArrayLikeBool_co,
45
+ _ArrayLikeBytes_co,
46
+ _ArrayLikeComplex_co,
47
+ _ArrayLikeFloat_co,
48
+ _ArrayLikeInt,
49
+ _ArrayLikeInt_co,
50
+ _ArrayLikeStr_co,
51
+ _ArrayLikeString_co,
52
+ _ArrayLikeTD64_co,
53
+ _DTypeLikeBool,
54
+ _IntLike_co,
55
+ _ScalarLike_co,
56
+ _Shape,
57
+ _ShapeLike,
58
+ )
59
+
60
+ __all__ = [
61
+ "MAError",
62
+ "MaskError",
63
+ "MaskType",
64
+ "MaskedArray",
65
+ "abs",
66
+ "absolute",
67
+ "add",
68
+ "all",
69
+ "allclose",
70
+ "allequal",
71
+ "alltrue",
72
+ "amax",
73
+ "amin",
74
+ "angle",
75
+ "anom",
76
+ "anomalies",
77
+ "any",
78
+ "append",
79
+ "arange",
80
+ "arccos",
81
+ "arccosh",
82
+ "arcsin",
83
+ "arcsinh",
84
+ "arctan",
85
+ "arctan2",
86
+ "arctanh",
87
+ "argmax",
88
+ "argmin",
89
+ "argsort",
90
+ "around",
91
+ "array",
92
+ "asanyarray",
93
+ "asarray",
94
+ "bitwise_and",
95
+ "bitwise_or",
96
+ "bitwise_xor",
97
+ "bool_",
98
+ "ceil",
99
+ "choose",
100
+ "clip",
101
+ "common_fill_value",
102
+ "compress",
103
+ "compressed",
104
+ "concatenate",
105
+ "conjugate",
106
+ "convolve",
107
+ "copy",
108
+ "correlate",
109
+ "cos",
110
+ "cosh",
111
+ "count",
112
+ "cumprod",
113
+ "cumsum",
114
+ "default_fill_value",
115
+ "diag",
116
+ "diagonal",
117
+ "diff",
118
+ "divide",
119
+ "empty",
120
+ "empty_like",
121
+ "equal",
122
+ "exp",
123
+ "expand_dims",
124
+ "fabs",
125
+ "filled",
126
+ "fix_invalid",
127
+ "flatten_mask",
128
+ "flatten_structured_array",
129
+ "floor",
130
+ "floor_divide",
131
+ "fmod",
132
+ "frombuffer",
133
+ "fromflex",
134
+ "fromfunction",
135
+ "getdata",
136
+ "getmask",
137
+ "getmaskarray",
138
+ "greater",
139
+ "greater_equal",
140
+ "harden_mask",
141
+ "hypot",
142
+ "identity",
143
+ "ids",
144
+ "indices",
145
+ "inner",
146
+ "innerproduct",
147
+ "isMA",
148
+ "isMaskedArray",
149
+ "is_mask",
150
+ "is_masked",
151
+ "isarray",
152
+ "left_shift",
153
+ "less",
154
+ "less_equal",
155
+ "log",
156
+ "log2",
157
+ "log10",
158
+ "logical_and",
159
+ "logical_not",
160
+ "logical_or",
161
+ "logical_xor",
162
+ "make_mask",
163
+ "make_mask_descr",
164
+ "make_mask_none",
165
+ "mask_or",
166
+ "masked",
167
+ "masked_array",
168
+ "masked_equal",
169
+ "masked_greater",
170
+ "masked_greater_equal",
171
+ "masked_inside",
172
+ "masked_invalid",
173
+ "masked_less",
174
+ "masked_less_equal",
175
+ "masked_not_equal",
176
+ "masked_object",
177
+ "masked_outside",
178
+ "masked_print_option",
179
+ "masked_singleton",
180
+ "masked_values",
181
+ "masked_where",
182
+ "max",
183
+ "maximum",
184
+ "maximum_fill_value",
185
+ "mean",
186
+ "min",
187
+ "minimum",
188
+ "minimum_fill_value",
189
+ "mod",
190
+ "multiply",
191
+ "mvoid",
192
+ "ndim",
193
+ "negative",
194
+ "nomask",
195
+ "nonzero",
196
+ "not_equal",
197
+ "ones",
198
+ "ones_like",
199
+ "outer",
200
+ "outerproduct",
201
+ "power",
202
+ "prod",
203
+ "product",
204
+ "ptp",
205
+ "put",
206
+ "putmask",
207
+ "ravel",
208
+ "remainder",
209
+ "repeat",
210
+ "reshape",
211
+ "resize",
212
+ "right_shift",
213
+ "round",
214
+ "round_",
215
+ "set_fill_value",
216
+ "shape",
217
+ "sin",
218
+ "sinh",
219
+ "size",
220
+ "soften_mask",
221
+ "sometrue",
222
+ "sort",
223
+ "sqrt",
224
+ "squeeze",
225
+ "std",
226
+ "subtract",
227
+ "sum",
228
+ "swapaxes",
229
+ "take",
230
+ "tan",
231
+ "tanh",
232
+ "trace",
233
+ "transpose",
234
+ "true_divide",
235
+ "var",
236
+ "where",
237
+ "zeros",
238
+ "zeros_like",
239
+ ]
240
+
241
+ _ShapeT = TypeVar("_ShapeT", bound=_Shape)
242
+ _ShapeT_co = TypeVar("_ShapeT_co", bound=_Shape, default=_AnyShape, covariant=True)
243
+ _DTypeT = TypeVar("_DTypeT", bound=dtype)
244
+ _DTypeT_co = TypeVar("_DTypeT_co", bound=dtype, default=dtype, covariant=True)
245
+ _ArrayT = TypeVar("_ArrayT", bound=ndarray[Any, Any])
246
+ _ScalarT = TypeVar("_ScalarT", bound=generic)
247
+ _ScalarT_co = TypeVar("_ScalarT_co", bound=generic, covariant=True)
248
+ # A subset of `MaskedArray` that can be parametrized w.r.t. `np.generic`
249
+ _MaskedArray: TypeAlias = MaskedArray[_AnyShape, dtype[_ScalarT]]
250
+ _Array1D: TypeAlias = np.ndarray[tuple[int], np.dtype[_ScalarT]]
251
+
252
+ MaskType = bool_
253
+ nomask: bool_[Literal[False]]
254
+
255
+ class MaskedArrayFutureWarning(FutureWarning): ...
256
+ class MAError(Exception): ...
257
+ class MaskError(MAError): ...
258
+
259
+ def default_fill_value(obj): ...
260
+ def minimum_fill_value(obj): ...
261
+ def maximum_fill_value(obj): ...
262
+ def set_fill_value(a, fill_value): ...
263
+ def common_fill_value(a, b): ...
264
+ @overload
265
+ def filled(a: ndarray[_ShapeT_co, _DTypeT_co], fill_value: _ScalarLike_co | None = None) -> ndarray[_ShapeT_co, _DTypeT_co]: ...
266
+ @overload
267
+ def filled(a: _ArrayLike[_ScalarT_co], fill_value: _ScalarLike_co | None = None) -> NDArray[_ScalarT_co]: ...
268
+ @overload
269
+ def filled(a: ArrayLike, fill_value: _ScalarLike_co | None = None) -> NDArray[Any]: ...
270
+ def getdata(a, subok=...): ...
271
+ get_data = getdata
272
+
273
+ def fix_invalid(a, mask=..., copy=..., fill_value=...): ...
274
+
275
+ class _MaskedUFunc:
276
+ f: Any
277
+ __doc__: Any
278
+ __name__: Any
279
+ def __init__(self, ufunc): ...
280
+
281
+ class _MaskedUnaryOperation(_MaskedUFunc):
282
+ fill: Any
283
+ domain: Any
284
+ def __init__(self, mufunc, fill=..., domain=...): ...
285
+ def __call__(self, a, *args, **kwargs): ...
286
+
287
+ class _MaskedBinaryOperation(_MaskedUFunc):
288
+ fillx: Any
289
+ filly: Any
290
+ def __init__(self, mbfunc, fillx=..., filly=...): ...
291
+ def __call__(self, a, b, *args, **kwargs): ...
292
+ def reduce(self, target, axis=..., dtype=...): ...
293
+ def outer(self, a, b): ...
294
+ def accumulate(self, target, axis=...): ...
295
+
296
+ class _DomainedBinaryOperation(_MaskedUFunc):
297
+ domain: Any
298
+ fillx: Any
299
+ filly: Any
300
+ def __init__(self, dbfunc, domain, fillx=..., filly=...): ...
301
+ def __call__(self, a, b, *args, **kwargs): ...
302
+
303
+ exp: _MaskedUnaryOperation
304
+ conjugate: _MaskedUnaryOperation
305
+ sin: _MaskedUnaryOperation
306
+ cos: _MaskedUnaryOperation
307
+ arctan: _MaskedUnaryOperation
308
+ arcsinh: _MaskedUnaryOperation
309
+ sinh: _MaskedUnaryOperation
310
+ cosh: _MaskedUnaryOperation
311
+ tanh: _MaskedUnaryOperation
312
+ abs: _MaskedUnaryOperation
313
+ absolute: _MaskedUnaryOperation
314
+ angle: _MaskedUnaryOperation
315
+ fabs: _MaskedUnaryOperation
316
+ negative: _MaskedUnaryOperation
317
+ floor: _MaskedUnaryOperation
318
+ ceil: _MaskedUnaryOperation
319
+ around: _MaskedUnaryOperation
320
+ logical_not: _MaskedUnaryOperation
321
+ sqrt: _MaskedUnaryOperation
322
+ log: _MaskedUnaryOperation
323
+ log2: _MaskedUnaryOperation
324
+ log10: _MaskedUnaryOperation
325
+ tan: _MaskedUnaryOperation
326
+ arcsin: _MaskedUnaryOperation
327
+ arccos: _MaskedUnaryOperation
328
+ arccosh: _MaskedUnaryOperation
329
+ arctanh: _MaskedUnaryOperation
330
+
331
+ add: _MaskedBinaryOperation
332
+ subtract: _MaskedBinaryOperation
333
+ multiply: _MaskedBinaryOperation
334
+ arctan2: _MaskedBinaryOperation
335
+ equal: _MaskedBinaryOperation
336
+ not_equal: _MaskedBinaryOperation
337
+ less_equal: _MaskedBinaryOperation
338
+ greater_equal: _MaskedBinaryOperation
339
+ less: _MaskedBinaryOperation
340
+ greater: _MaskedBinaryOperation
341
+ logical_and: _MaskedBinaryOperation
342
+ def alltrue(target: ArrayLike, axis: SupportsIndex | None = 0, dtype: _DTypeLikeBool | None = None) -> Incomplete: ...
343
+ logical_or: _MaskedBinaryOperation
344
+ def sometrue(target: ArrayLike, axis: SupportsIndex | None = 0, dtype: _DTypeLikeBool | None = None) -> Incomplete: ...
345
+ logical_xor: _MaskedBinaryOperation
346
+ bitwise_and: _MaskedBinaryOperation
347
+ bitwise_or: _MaskedBinaryOperation
348
+ bitwise_xor: _MaskedBinaryOperation
349
+ hypot: _MaskedBinaryOperation
350
+
351
+ divide: _DomainedBinaryOperation
352
+ true_divide: _DomainedBinaryOperation
353
+ floor_divide: _DomainedBinaryOperation
354
+ remainder: _DomainedBinaryOperation
355
+ fmod: _DomainedBinaryOperation
356
+ mod: _DomainedBinaryOperation
357
+
358
+ def make_mask_descr(ndtype): ...
359
+
360
+ @overload
361
+ def getmask(a: _ScalarLike_co) -> bool_: ...
362
+ @overload
363
+ def getmask(a: MaskedArray[_ShapeT_co, Any]) -> np.ndarray[_ShapeT_co, dtype[bool_]] | bool_: ...
364
+ @overload
365
+ def getmask(a: ArrayLike) -> NDArray[bool_] | bool_: ...
366
+
367
+ get_mask = getmask
368
+
369
+ def getmaskarray(arr): ...
370
+
371
+ # It's sufficient for `m` to have dtype with type: `type[np.bool_]`,
372
+ # which isn't necessarily a ndarray. Please open an issue if this causes issues.
373
+ def is_mask(m: object) -> TypeIs[NDArray[bool_]]: ...
374
+
375
+ def make_mask(m, copy=..., shrink=..., dtype=...): ...
376
+ def make_mask_none(newshape, dtype=...): ...
377
+ def mask_or(m1, m2, copy=..., shrink=...): ...
378
+ def flatten_mask(mask): ...
379
+ def masked_where(condition, a, copy=...): ...
380
+ def masked_greater(x, value, copy=...): ...
381
+ def masked_greater_equal(x, value, copy=...): ...
382
+ def masked_less(x, value, copy=...): ...
383
+ def masked_less_equal(x, value, copy=...): ...
384
+ def masked_not_equal(x, value, copy=...): ...
385
+ def masked_equal(x, value, copy=...): ...
386
+ def masked_inside(x, v1, v2, copy=...): ...
387
+ def masked_outside(x, v1, v2, copy=...): ...
388
+ def masked_object(x, value, copy=..., shrink=...): ...
389
+ def masked_values(x, value, rtol=..., atol=..., copy=..., shrink=...): ...
390
+ def masked_invalid(a, copy=...): ...
391
+
392
+ class _MaskedPrintOption:
393
+ def __init__(self, display): ...
394
+ def display(self): ...
395
+ def set_display(self, s): ...
396
+ def enabled(self): ...
397
+ def enable(self, shrink=...): ...
398
+
399
+ masked_print_option: _MaskedPrintOption
400
+
401
+ def flatten_structured_array(a): ...
402
+
403
+ class MaskedIterator:
404
+ ma: Any
405
+ dataiter: Any
406
+ maskiter: Any
407
+ def __init__(self, ma): ...
408
+ def __iter__(self): ...
409
+ def __getitem__(self, indx): ...
410
+ def __setitem__(self, index, value): ...
411
+ def __next__(self): ...
412
+
413
+ class MaskedArray(ndarray[_ShapeT_co, _DTypeT_co]):
414
+ __array_priority__: Any
415
+ def __new__(cls, data=..., mask=..., dtype=..., copy=..., subok=..., ndmin=..., fill_value=..., keep_mask=..., hard_mask=..., shrink=..., order=...): ...
416
+ def __array_finalize__(self, obj): ...
417
+ def __array_wrap__(self, obj, context=..., return_scalar=...): ...
418
+ def view(self, dtype=..., type=..., fill_value=...): ...
419
+ def __getitem__(self, indx): ...
420
+ def __setitem__(self, indx, value): ...
421
+ @property
422
+ def shape(self) -> _ShapeT_co: ...
423
+ @shape.setter
424
+ def shape(self: MaskedArray[_ShapeT, Any], shape: _ShapeT, /) -> None: ...
425
+ def __setmask__(self, mask: _ArrayLikeBool_co, copy: bool = False) -> None: ...
426
+ @property
427
+ def mask(self) -> NDArray[MaskType] | MaskType: ...
428
+ @mask.setter
429
+ def mask(self, value: _ArrayLikeBool_co, /) -> None: ...
430
+ @property
431
+ def recordmask(self): ...
432
+ @recordmask.setter
433
+ def recordmask(self, mask): ...
434
+ def harden_mask(self) -> Self: ...
435
+ def soften_mask(self) -> Self: ...
436
+ @property
437
+ def hardmask(self) -> bool: ...
438
+ def unshare_mask(self) -> Self: ...
439
+ @property
440
+ def sharedmask(self) -> bool: ...
441
+ def shrink_mask(self) -> Self: ...
442
+ @property
443
+ def baseclass(self) -> type[NDArray[Any]]: ...
444
+ data: Any
445
+ @property
446
+ def flat(self): ...
447
+ @flat.setter
448
+ def flat(self, value): ...
449
+ @property
450
+ def fill_value(self): ...
451
+ @fill_value.setter
452
+ def fill_value(self, value=...): ...
453
+ get_fill_value: Any
454
+ set_fill_value: Any
455
+ def filled(self, /, fill_value: _ScalarLike_co | None = None) -> ndarray[_ShapeT_co, _DTypeT_co]: ...
456
+ def compressed(self) -> ndarray[tuple[int], _DTypeT_co]: ...
457
+ def compress(self, condition, axis=..., out=...): ...
458
+ def __eq__(self, other): ...
459
+ def __ne__(self, other): ...
460
+ def __ge__(self, other: ArrayLike, /) -> _MaskedArray[bool_]: ... # type: ignore[override]
461
+ def __gt__(self, other: ArrayLike, /) -> _MaskedArray[bool_]: ... # type: ignore[override]
462
+ def __le__(self, other: ArrayLike, /) -> _MaskedArray[bool_]: ... # type: ignore[override]
463
+ def __lt__(self, other: ArrayLike, /) -> _MaskedArray[bool_]: ... # type: ignore[override]
464
+ def __add__(self, other): ...
465
+ def __radd__(self, other): ...
466
+ def __sub__(self, other): ...
467
+ def __rsub__(self, other): ...
468
+ def __mul__(self, other): ...
469
+ def __rmul__(self, other): ...
470
+ def __truediv__(self, other): ...
471
+ def __rtruediv__(self, other): ...
472
+ def __floordiv__(self, other): ...
473
+ def __rfloordiv__(self, other): ...
474
+ def __pow__(self, other, mod: None = None, /): ...
475
+ def __rpow__(self, other, mod: None = None, /): ...
476
+
477
+ # Keep in sync with `ndarray.__iadd__`
478
+ @overload
479
+ def __iadd__(
480
+ self: _MaskedArray[np.bool], other: _ArrayLikeBool_co, /
481
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
482
+ @overload
483
+ def __iadd__(self: _MaskedArray[integer], other: _ArrayLikeInt_co, /) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
484
+ @overload
485
+ def __iadd__(
486
+ self: _MaskedArray[floating], other: _ArrayLikeFloat_co, /
487
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
488
+ @overload
489
+ def __iadd__(
490
+ self: _MaskedArray[complexfloating], other: _ArrayLikeComplex_co, /
491
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
492
+ @overload
493
+ def __iadd__(
494
+ self: _MaskedArray[timedelta64 | datetime64], other: _ArrayLikeTD64_co, /
495
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
496
+ @overload
497
+ def __iadd__(self: _MaskedArray[bytes_], other: _ArrayLikeBytes_co, /) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
498
+ @overload
499
+ def __iadd__(
500
+ self: MaskedArray[Any, dtype[str_] | dtypes.StringDType],
501
+ other: _ArrayLikeStr_co | _ArrayLikeString_co,
502
+ /,
503
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
504
+ @overload
505
+ def __iadd__(
506
+ self: _MaskedArray[object_], other: Any, /
507
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
508
+
509
+ # Keep in sync with `ndarray.__isub__`
510
+ @overload
511
+ def __isub__(self: _MaskedArray[integer], other: _ArrayLikeInt_co, /) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
512
+ @overload
513
+ def __isub__(
514
+ self: _MaskedArray[floating], other: _ArrayLikeFloat_co, /
515
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
516
+ @overload
517
+ def __isub__(
518
+ self: _MaskedArray[complexfloating], other: _ArrayLikeComplex_co, /
519
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
520
+ @overload
521
+ def __isub__(
522
+ self: _MaskedArray[timedelta64 | datetime64], other: _ArrayLikeTD64_co, /
523
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
524
+ @overload
525
+ def __isub__(
526
+ self: _MaskedArray[object_], other: Any, /
527
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
528
+
529
+ # Keep in sync with `ndarray.__imul__`
530
+ @overload
531
+ def __imul__(
532
+ self: _MaskedArray[np.bool], other: _ArrayLikeBool_co, /
533
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
534
+ @overload
535
+ def __imul__(
536
+ self: MaskedArray[Any, dtype[integer] | dtype[character] | dtypes.StringDType], other: _ArrayLikeInt_co, /
537
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
538
+ @overload
539
+ def __imul__(
540
+ self: _MaskedArray[floating | timedelta64], other: _ArrayLikeFloat_co, /
541
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
542
+ @overload
543
+ def __imul__(
544
+ self: _MaskedArray[complexfloating], other: _ArrayLikeComplex_co, /
545
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
546
+ @overload
547
+ def __imul__(
548
+ self: _MaskedArray[object_], other: Any, /
549
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
550
+
551
+ # Keep in sync with `ndarray.__ifloordiv__`
552
+ @overload
553
+ def __ifloordiv__(self: _MaskedArray[integer], other: _ArrayLikeInt_co, /) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
554
+ @overload
555
+ def __ifloordiv__(
556
+ self: _MaskedArray[floating | timedelta64], other: _ArrayLikeFloat_co, /
557
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
558
+ @overload
559
+ def __ifloordiv__(
560
+ self: _MaskedArray[object_], other: Any, /
561
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
562
+
563
+ # Keep in sync with `ndarray.__itruediv__`
564
+ @overload
565
+ def __itruediv__(
566
+ self: _MaskedArray[floating | timedelta64], other: _ArrayLikeFloat_co, /
567
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
568
+ @overload
569
+ def __itruediv__(
570
+ self: _MaskedArray[complexfloating],
571
+ other: _ArrayLikeComplex_co,
572
+ /,
573
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
574
+ @overload
575
+ def __itruediv__(
576
+ self: _MaskedArray[object_], other: Any, /
577
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
578
+
579
+ # Keep in sync with `ndarray.__ipow__`
580
+ @overload
581
+ def __ipow__(self: _MaskedArray[integer], other: _ArrayLikeInt_co, /) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
582
+ @overload
583
+ def __ipow__(
584
+ self: _MaskedArray[floating], other: _ArrayLikeFloat_co, /
585
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
586
+ @overload
587
+ def __ipow__(
588
+ self: _MaskedArray[complexfloating], other: _ArrayLikeComplex_co, /
589
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
590
+ @overload
591
+ def __ipow__(
592
+ self: _MaskedArray[object_], other: Any, /
593
+ ) -> MaskedArray[_ShapeT_co, _DTypeT_co]: ...
594
+
595
+ #
596
+ @property # type: ignore[misc]
597
+ def imag(self: _HasDTypeWithRealAndImag[object, _ScalarT], /) -> MaskedArray[_ShapeT_co, dtype[_ScalarT]]: ...
598
+ get_imag: Any
599
+ @property # type: ignore[misc]
600
+ def real(self: _HasDTypeWithRealAndImag[_ScalarT, object], /) -> MaskedArray[_ShapeT_co, dtype[_ScalarT]]: ...
601
+ get_real: Any
602
+
603
+ # keep in sync with `np.ma.count`
604
+ @overload
605
+ def count(self, axis: None = None, keepdims: Literal[False] | _NoValueType = ...) -> int: ...
606
+ @overload
607
+ def count(self, axis: _ShapeLike, keepdims: bool | _NoValueType = ...) -> NDArray[int_]: ...
608
+ @overload
609
+ def count(self, axis: _ShapeLike | None = ..., *, keepdims: Literal[True]) -> NDArray[int_]: ...
610
+ @overload
611
+ def count(self, axis: _ShapeLike | None, keepdims: Literal[True]) -> NDArray[int_]: ...
612
+
613
+ def ravel(self, order: _OrderKACF = "C") -> MaskedArray[tuple[int], _DTypeT_co]: ...
614
+ def reshape(self, *s, **kwargs): ...
615
+ def resize(self, newshape, refcheck=..., order=...): ...
616
+ def put(self, indices: _ArrayLikeInt_co, values: ArrayLike, mode: _ModeKind = "raise") -> None: ...
617
+ def ids(self) -> tuple[int, int]: ...
618
+ def iscontiguous(self) -> bool: ...
619
+
620
+ @overload
621
+ def all(
622
+ self,
623
+ axis: None = None,
624
+ out: None = None,
625
+ keepdims: Literal[False] | _NoValueType = ...,
626
+ ) -> bool_: ...
627
+ @overload
628
+ def all(
629
+ self,
630
+ axis: _ShapeLike | None = None,
631
+ out: None = None,
632
+ *,
633
+ keepdims: Literal[True],
634
+ ) -> _MaskedArray[bool_]: ...
635
+ @overload
636
+ def all(
637
+ self,
638
+ axis: _ShapeLike | None,
639
+ out: None,
640
+ keepdims: Literal[True],
641
+ ) -> _MaskedArray[bool_]: ...
642
+ @overload
643
+ def all(
644
+ self,
645
+ axis: _ShapeLike | None = None,
646
+ out: None = None,
647
+ keepdims: bool | _NoValueType = ...,
648
+ ) -> bool_ | _MaskedArray[bool_]: ...
649
+ @overload
650
+ def all(
651
+ self,
652
+ axis: _ShapeLike | None = None,
653
+ *,
654
+ out: _ArrayT,
655
+ keepdims: bool | _NoValueType = ...,
656
+ ) -> _ArrayT: ...
657
+ @overload
658
+ def all(
659
+ self,
660
+ axis: _ShapeLike | None,
661
+ out: _ArrayT,
662
+ keepdims: bool | _NoValueType = ...,
663
+ ) -> _ArrayT: ...
664
+
665
+ @overload
666
+ def any(
667
+ self,
668
+ axis: None = None,
669
+ out: None = None,
670
+ keepdims: Literal[False] | _NoValueType = ...,
671
+ ) -> bool_: ...
672
+ @overload
673
+ def any(
674
+ self,
675
+ axis: _ShapeLike | None = None,
676
+ out: None = None,
677
+ *,
678
+ keepdims: Literal[True],
679
+ ) -> _MaskedArray[bool_]: ...
680
+ @overload
681
+ def any(
682
+ self,
683
+ axis: _ShapeLike | None,
684
+ out: None,
685
+ keepdims: Literal[True],
686
+ ) -> _MaskedArray[bool_]: ...
687
+ @overload
688
+ def any(
689
+ self,
690
+ axis: _ShapeLike | None = None,
691
+ out: None = None,
692
+ keepdims: bool | _NoValueType = ...,
693
+ ) -> bool_ | _MaskedArray[bool_]: ...
694
+ @overload
695
+ def any(
696
+ self,
697
+ axis: _ShapeLike | None = None,
698
+ *,
699
+ out: _ArrayT,
700
+ keepdims: bool | _NoValueType = ...,
701
+ ) -> _ArrayT: ...
702
+ @overload
703
+ def any(
704
+ self,
705
+ axis: _ShapeLike | None,
706
+ out: _ArrayT,
707
+ keepdims: bool | _NoValueType = ...,
708
+ ) -> _ArrayT: ...
709
+
710
+ def nonzero(self) -> tuple[_Array1D[intp], *tuple[_Array1D[intp], ...]]: ...
711
+ def trace(self, offset=..., axis1=..., axis2=..., dtype=..., out=...): ...
712
+ def dot(self, b, out=..., strict=...): ...
713
+ def sum(self, axis=..., dtype=..., out=..., keepdims=...): ...
714
+ def cumsum(self, axis=..., dtype=..., out=...): ...
715
+ def prod(self, axis=..., dtype=..., out=..., keepdims=...): ...
716
+ product: Any
717
+ def cumprod(self, axis=..., dtype=..., out=...): ...
718
+ def mean(self, axis=..., dtype=..., out=..., keepdims=...): ...
719
+ def anom(self, axis=..., dtype=...): ...
720
+ def var(self, axis=..., dtype=..., out=..., ddof=..., keepdims=...): ...
721
+ def std(self, axis=..., dtype=..., out=..., ddof=..., keepdims=...): ...
722
+ def round(self, decimals=..., out=...): ...
723
+ def argsort(self, axis=..., kind=..., order=..., endwith=..., fill_value=..., *, stable=...): ...
724
+
725
+ # Keep in-sync with np.ma.argmin
726
+ @overload # type: ignore[override]
727
+ def argmin(
728
+ self,
729
+ axis: None = None,
730
+ fill_value: _ScalarLike_co | None = None,
731
+ out: None = None,
732
+ *,
733
+ keepdims: Literal[False] | _NoValueType = ...,
734
+ ) -> intp: ...
735
+ @overload
736
+ def argmin(
737
+ self,
738
+ axis: SupportsIndex | None = None,
739
+ fill_value: _ScalarLike_co | None = None,
740
+ out: None = None,
741
+ *,
742
+ keepdims: bool | _NoValueType = ...,
743
+ ) -> Any: ...
744
+ @overload
745
+ def argmin(
746
+ self,
747
+ axis: SupportsIndex | None = None,
748
+ fill_value: _ScalarLike_co | None = None,
749
+ *,
750
+ out: _ArrayT,
751
+ keepdims: bool | _NoValueType = ...,
752
+ ) -> _ArrayT: ...
753
+ @overload
754
+ def argmin(
755
+ self,
756
+ axis: SupportsIndex | None,
757
+ fill_value: _ScalarLike_co | None,
758
+ out: _ArrayT,
759
+ *,
760
+ keepdims: bool | _NoValueType = ...,
761
+ ) -> _ArrayT: ...
762
+
763
+ # Keep in-sync with np.ma.argmax
764
+ @overload # type: ignore[override]
765
+ def argmax(
766
+ self,
767
+ axis: None = None,
768
+ fill_value: _ScalarLike_co | None = None,
769
+ out: None = None,
770
+ *,
771
+ keepdims: Literal[False] | _NoValueType = ...,
772
+ ) -> intp: ...
773
+ @overload
774
+ def argmax(
775
+ self,
776
+ axis: SupportsIndex | None = None,
777
+ fill_value: _ScalarLike_co | None = None,
778
+ out: None = None,
779
+ *,
780
+ keepdims: bool | _NoValueType = ...,
781
+ ) -> Any: ...
782
+ @overload
783
+ def argmax(
784
+ self,
785
+ axis: SupportsIndex | None = None,
786
+ fill_value: _ScalarLike_co | None = None,
787
+ *,
788
+ out: _ArrayT,
789
+ keepdims: bool | _NoValueType = ...,
790
+ ) -> _ArrayT: ...
791
+ @overload
792
+ def argmax(
793
+ self,
794
+ axis: SupportsIndex | None,
795
+ fill_value: _ScalarLike_co | None,
796
+ out: _ArrayT,
797
+ *,
798
+ keepdims: bool | _NoValueType = ...,
799
+ ) -> _ArrayT: ...
800
+
801
+ #
802
+ def sort( # type: ignore[override]
803
+ self,
804
+ axis: SupportsIndex = -1,
805
+ kind: _SortKind | None = None,
806
+ order: str | Sequence[str] | None = None,
807
+ endwith: bool | None = True,
808
+ fill_value: _ScalarLike_co | None = None,
809
+ *,
810
+ stable: Literal[False] | None = False,
811
+ ) -> None: ...
812
+
813
+ #
814
+ @overload # type: ignore[override]
815
+ def min(
816
+ self: _MaskedArray[_ScalarT],
817
+ axis: None = None,
818
+ out: None = None,
819
+ fill_value: _ScalarLike_co | None = None,
820
+ keepdims: Literal[False] | _NoValueType = ...,
821
+ ) -> _ScalarT: ...
822
+ @overload
823
+ def min(
824
+ self,
825
+ axis: _ShapeLike | None = None,
826
+ out: None = None,
827
+ fill_value: _ScalarLike_co | None = None,
828
+ keepdims: bool | _NoValueType = ...
829
+ ) -> Any: ...
830
+ @overload
831
+ def min(
832
+ self,
833
+ axis: _ShapeLike | None,
834
+ out: _ArrayT,
835
+ fill_value: _ScalarLike_co | None = None,
836
+ keepdims: bool | _NoValueType = ...,
837
+ ) -> _ArrayT: ...
838
+ @overload
839
+ def min(
840
+ self,
841
+ axis: _ShapeLike | None = None,
842
+ *,
843
+ out: _ArrayT,
844
+ fill_value: _ScalarLike_co | None = None,
845
+ keepdims: bool | _NoValueType = ...,
846
+ ) -> _ArrayT: ...
847
+
848
+ #
849
+ @overload # type: ignore[override]
850
+ def max(
851
+ self: _MaskedArray[_ScalarT],
852
+ axis: None = None,
853
+ out: None = None,
854
+ fill_value: _ScalarLike_co | None = None,
855
+ keepdims: Literal[False] | _NoValueType = ...,
856
+ ) -> _ScalarT: ...
857
+ @overload
858
+ def max(
859
+ self,
860
+ axis: _ShapeLike | None = None,
861
+ out: None = None,
862
+ fill_value: _ScalarLike_co | None = None,
863
+ keepdims: bool | _NoValueType = ...
864
+ ) -> Any: ...
865
+ @overload
866
+ def max(
867
+ self,
868
+ axis: _ShapeLike | None,
869
+ out: _ArrayT,
870
+ fill_value: _ScalarLike_co | None = None,
871
+ keepdims: bool | _NoValueType = ...,
872
+ ) -> _ArrayT: ...
873
+ @overload
874
+ def max(
875
+ self,
876
+ axis: _ShapeLike | None = None,
877
+ *,
878
+ out: _ArrayT,
879
+ fill_value: _ScalarLike_co | None = None,
880
+ keepdims: bool | _NoValueType = ...,
881
+ ) -> _ArrayT: ...
882
+
883
+ #
884
+ @overload
885
+ def ptp(
886
+ self: _MaskedArray[_ScalarT],
887
+ axis: None = None,
888
+ out: None = None,
889
+ fill_value: _ScalarLike_co | None = None,
890
+ keepdims: Literal[False] = False,
891
+ ) -> _ScalarT: ...
892
+ @overload
893
+ def ptp(
894
+ self,
895
+ axis: _ShapeLike | None = None,
896
+ out: None = None,
897
+ fill_value: _ScalarLike_co | None = None,
898
+ keepdims: bool = False,
899
+ ) -> Any: ...
900
+ @overload
901
+ def ptp(
902
+ self,
903
+ axis: _ShapeLike | None,
904
+ out: _ArrayT,
905
+ fill_value: _ScalarLike_co | None = None,
906
+ keepdims: bool = False,
907
+ ) -> _ArrayT: ...
908
+ @overload
909
+ def ptp(
910
+ self,
911
+ axis: _ShapeLike | None = None,
912
+ *,
913
+ out: _ArrayT,
914
+ fill_value: _ScalarLike_co | None = None,
915
+ keepdims: bool = False,
916
+ ) -> _ArrayT: ...
917
+
918
+ #
919
+ @overload
920
+ def partition(
921
+ self,
922
+ /,
923
+ kth: _ArrayLikeInt,
924
+ axis: SupportsIndex = -1,
925
+ kind: _PartitionKind = "introselect",
926
+ order: None = None
927
+ ) -> None: ...
928
+ @overload
929
+ def partition(
930
+ self: _MaskedArray[np.void],
931
+ /,
932
+ kth: _ArrayLikeInt,
933
+ axis: SupportsIndex = -1,
934
+ kind: _PartitionKind = "introselect",
935
+ order: str | Sequence[str] | None = None,
936
+ ) -> None: ...
937
+
938
+ #
939
+ @overload
940
+ def argpartition(
941
+ self,
942
+ /,
943
+ kth: _ArrayLikeInt,
944
+ axis: SupportsIndex | None = -1,
945
+ kind: _PartitionKind = "introselect",
946
+ order: None = None,
947
+ ) -> _MaskedArray[intp]: ...
948
+ @overload
949
+ def argpartition(
950
+ self: _MaskedArray[np.void],
951
+ /,
952
+ kth: _ArrayLikeInt,
953
+ axis: SupportsIndex | None = -1,
954
+ kind: _PartitionKind = "introselect",
955
+ order: str | Sequence[str] | None = None,
956
+ ) -> _MaskedArray[intp]: ...
957
+
958
+ # Keep in-sync with np.ma.take
959
+ @overload
960
+ def take( # type: ignore[overload-overlap]
961
+ self: _MaskedArray[_ScalarT],
962
+ indices: _IntLike_co,
963
+ axis: None = None,
964
+ out: None = None,
965
+ mode: _ModeKind = 'raise'
966
+ ) -> _ScalarT: ...
967
+ @overload
968
+ def take(
969
+ self: _MaskedArray[_ScalarT],
970
+ indices: _ArrayLikeInt_co,
971
+ axis: SupportsIndex | None = None,
972
+ out: None = None,
973
+ mode: _ModeKind = 'raise',
974
+ ) -> _MaskedArray[_ScalarT]: ...
975
+ @overload
976
+ def take(
977
+ self,
978
+ indices: _ArrayLikeInt_co,
979
+ axis: SupportsIndex | None,
980
+ out: _ArrayT,
981
+ mode: _ModeKind = 'raise',
982
+ ) -> _ArrayT: ...
983
+ @overload
984
+ def take(
985
+ self,
986
+ indices: _ArrayLikeInt_co,
987
+ axis: SupportsIndex | None = None,
988
+ *,
989
+ out: _ArrayT,
990
+ mode: _ModeKind = 'raise',
991
+ ) -> _ArrayT: ...
992
+
993
+ copy: Any
994
+ diagonal: Any
995
+ flatten: Any
996
+
997
+ @overload
998
+ def repeat(
999
+ self,
1000
+ repeats: _ArrayLikeInt_co,
1001
+ axis: None = None,
1002
+ ) -> MaskedArray[tuple[int], _DTypeT_co]: ...
1003
+ @overload
1004
+ def repeat(
1005
+ self,
1006
+ repeats: _ArrayLikeInt_co,
1007
+ axis: SupportsIndex,
1008
+ ) -> MaskedArray[_AnyShape, _DTypeT_co]: ...
1009
+
1010
+ squeeze: Any
1011
+
1012
+ def swapaxes(
1013
+ self,
1014
+ axis1: SupportsIndex,
1015
+ axis2: SupportsIndex,
1016
+ /
1017
+ ) -> MaskedArray[_AnyShape, _DTypeT_co]: ...
1018
+
1019
+ #
1020
+ def toflex(self) -> Incomplete: ...
1021
+ def torecords(self) -> Incomplete: ...
1022
+ def tolist(self, fill_value: Incomplete | None = None) -> Incomplete: ...
1023
+ def tobytes(self, /, fill_value: Incomplete | None = None, order: _OrderKACF = "C") -> bytes: ... # type: ignore[override]
1024
+ def tofile(self, /, fid: Incomplete, sep: str = "", format: str = "%s") -> Incomplete: ...
1025
+
1026
+ #
1027
+ def __reduce__(self): ...
1028
+ def __deepcopy__(self, memo=...): ...
1029
+
1030
+ # Keep `dtype` at the bottom to avoid name conflicts with `np.dtype`
1031
+ @property
1032
+ def dtype(self) -> _DTypeT_co: ...
1033
+ @dtype.setter
1034
+ def dtype(self: MaskedArray[_AnyShape, _DTypeT], dtype: _DTypeT, /) -> None: ...
1035
+
1036
+ class mvoid(MaskedArray[_ShapeT_co, _DTypeT_co]):
1037
+ def __new__(
1038
+ self, # pyright: ignore[reportSelfClsParameterName]
1039
+ data,
1040
+ mask=...,
1041
+ dtype=...,
1042
+ fill_value=...,
1043
+ hardmask=...,
1044
+ copy=...,
1045
+ subok=...,
1046
+ ): ...
1047
+ def __getitem__(self, indx): ...
1048
+ def __setitem__(self, indx, value): ...
1049
+ def __iter__(self): ...
1050
+ def __len__(self): ...
1051
+ def filled(self, fill_value=...): ...
1052
+ def tolist(self): ...
1053
+
1054
+ def isMaskedArray(x): ...
1055
+ isarray = isMaskedArray
1056
+ isMA = isMaskedArray
1057
+
1058
+ # 0D float64 array
1059
+ class MaskedConstant(MaskedArray[_AnyShape, dtype[float64]]):
1060
+ def __new__(cls): ...
1061
+ __class__: Any
1062
+ def __array_finalize__(self, obj): ...
1063
+ def __array_wrap__(self, obj, context=..., return_scalar=...): ...
1064
+ def __format__(self, format_spec): ...
1065
+ def __reduce__(self): ...
1066
+ def __iop__(self, other): ...
1067
+ __iadd__: Any
1068
+ __isub__: Any
1069
+ __imul__: Any
1070
+ __ifloordiv__: Any
1071
+ __itruediv__: Any
1072
+ __ipow__: Any
1073
+ def copy(self, *args, **kwargs): ...
1074
+ def __copy__(self): ...
1075
+ def __deepcopy__(self, memo): ...
1076
+ def __setattr__(self, attr, value): ...
1077
+
1078
+ masked: MaskedConstant
1079
+ masked_singleton: MaskedConstant
1080
+ masked_array = MaskedArray
1081
+
1082
+ def array(
1083
+ data,
1084
+ dtype=...,
1085
+ copy=...,
1086
+ order=...,
1087
+ mask=...,
1088
+ fill_value=...,
1089
+ keep_mask=...,
1090
+ hard_mask=...,
1091
+ shrink=...,
1092
+ subok=...,
1093
+ ndmin=...,
1094
+ ): ...
1095
+ def is_masked(x: object) -> bool: ...
1096
+
1097
+ class _extrema_operation(_MaskedUFunc):
1098
+ compare: Any
1099
+ fill_value_func: Any
1100
+ def __init__(self, ufunc, compare, fill_value): ...
1101
+ # NOTE: in practice `b` has a default value, but users should
1102
+ # explicitly provide a value here as the default is deprecated
1103
+ def __call__(self, a, b): ...
1104
+ def reduce(self, target, axis=...): ...
1105
+ def outer(self, a, b): ...
1106
+
1107
+ @overload
1108
+ def min(
1109
+ obj: _ArrayLike[_ScalarT],
1110
+ axis: None = None,
1111
+ out: None = None,
1112
+ fill_value: _ScalarLike_co | None = None,
1113
+ keepdims: Literal[False] | _NoValueType = ...,
1114
+ ) -> _ScalarT: ...
1115
+ @overload
1116
+ def min(
1117
+ obj: ArrayLike,
1118
+ axis: _ShapeLike | None = None,
1119
+ out: None = None,
1120
+ fill_value: _ScalarLike_co | None = None,
1121
+ keepdims: bool | _NoValueType = ...
1122
+ ) -> Any: ...
1123
+ @overload
1124
+ def min(
1125
+ obj: ArrayLike,
1126
+ axis: _ShapeLike | None,
1127
+ out: _ArrayT,
1128
+ fill_value: _ScalarLike_co | None = None,
1129
+ keepdims: bool | _NoValueType = ...,
1130
+ ) -> _ArrayT: ...
1131
+ @overload
1132
+ def min(
1133
+ obj: ArrayLike,
1134
+ axis: _ShapeLike | None = None,
1135
+ *,
1136
+ out: _ArrayT,
1137
+ fill_value: _ScalarLike_co | None = None,
1138
+ keepdims: bool | _NoValueType = ...,
1139
+ ) -> _ArrayT: ...
1140
+
1141
+ @overload
1142
+ def max(
1143
+ obj: _ArrayLike[_ScalarT],
1144
+ axis: None = None,
1145
+ out: None = None,
1146
+ fill_value: _ScalarLike_co | None = None,
1147
+ keepdims: Literal[False] | _NoValueType = ...,
1148
+ ) -> _ScalarT: ...
1149
+ @overload
1150
+ def max(
1151
+ obj: ArrayLike,
1152
+ axis: _ShapeLike | None = None,
1153
+ out: None = None,
1154
+ fill_value: _ScalarLike_co | None = None,
1155
+ keepdims: bool | _NoValueType = ...
1156
+ ) -> Any: ...
1157
+ @overload
1158
+ def max(
1159
+ obj: ArrayLike,
1160
+ axis: _ShapeLike | None,
1161
+ out: _ArrayT,
1162
+ fill_value: _ScalarLike_co | None = None,
1163
+ keepdims: bool | _NoValueType = ...,
1164
+ ) -> _ArrayT: ...
1165
+ @overload
1166
+ def max(
1167
+ obj: ArrayLike,
1168
+ axis: _ShapeLike | None = None,
1169
+ *,
1170
+ out: _ArrayT,
1171
+ fill_value: _ScalarLike_co | None = None,
1172
+ keepdims: bool | _NoValueType = ...,
1173
+ ) -> _ArrayT: ...
1174
+
1175
+ @overload
1176
+ def ptp(
1177
+ obj: _ArrayLike[_ScalarT],
1178
+ axis: None = None,
1179
+ out: None = None,
1180
+ fill_value: _ScalarLike_co | None = None,
1181
+ keepdims: Literal[False] | _NoValueType = ...,
1182
+ ) -> _ScalarT: ...
1183
+ @overload
1184
+ def ptp(
1185
+ obj: ArrayLike,
1186
+ axis: _ShapeLike | None = None,
1187
+ out: None = None,
1188
+ fill_value: _ScalarLike_co | None = None,
1189
+ keepdims: bool | _NoValueType = ...
1190
+ ) -> Any: ...
1191
+ @overload
1192
+ def ptp(
1193
+ obj: ArrayLike,
1194
+ axis: _ShapeLike | None,
1195
+ out: _ArrayT,
1196
+ fill_value: _ScalarLike_co | None = None,
1197
+ keepdims: bool | _NoValueType = ...,
1198
+ ) -> _ArrayT: ...
1199
+ @overload
1200
+ def ptp(
1201
+ obj: ArrayLike,
1202
+ axis: _ShapeLike | None = None,
1203
+ *,
1204
+ out: _ArrayT,
1205
+ fill_value: _ScalarLike_co | None = None,
1206
+ keepdims: bool | _NoValueType = ...,
1207
+ ) -> _ArrayT: ...
1208
+
1209
+ class _frommethod:
1210
+ __name__: Any
1211
+ __doc__: Any
1212
+ reversed: Any
1213
+ def __init__(self, methodname, reversed=...): ...
1214
+ def getdoc(self): ...
1215
+ def __call__(self, a, *args, **params): ...
1216
+
1217
+ all: _frommethod
1218
+ anomalies: _frommethod
1219
+ anom: _frommethod
1220
+ any: _frommethod
1221
+ compress: _frommethod
1222
+ cumprod: _frommethod
1223
+ cumsum: _frommethod
1224
+ copy: _frommethod
1225
+ diagonal: _frommethod
1226
+ harden_mask: _frommethod
1227
+ ids: _frommethod
1228
+ mean: _frommethod
1229
+ nonzero: _frommethod
1230
+ prod: _frommethod
1231
+ product: _frommethod
1232
+ ravel: _frommethod
1233
+ repeat: _frommethod
1234
+ soften_mask: _frommethod
1235
+ std: _frommethod
1236
+ sum: _frommethod
1237
+ swapaxes: _frommethod
1238
+ trace: _frommethod
1239
+ var: _frommethod
1240
+
1241
+ @overload
1242
+ def count(self: ArrayLike, axis: None = None, keepdims: Literal[False] | _NoValueType = ...) -> int: ...
1243
+ @overload
1244
+ def count(self: ArrayLike, axis: _ShapeLike, keepdims: bool | _NoValueType = ...) -> NDArray[int_]: ...
1245
+ @overload
1246
+ def count(self: ArrayLike, axis: _ShapeLike | None = ..., *, keepdims: Literal[True]) -> NDArray[int_]: ...
1247
+ @overload
1248
+ def count(self: ArrayLike, axis: _ShapeLike | None, keepdims: Literal[True]) -> NDArray[int_]: ...
1249
+
1250
+ @overload
1251
+ def argmin(
1252
+ self: ArrayLike,
1253
+ axis: None = None,
1254
+ fill_value: _ScalarLike_co | None = None,
1255
+ out: None = None,
1256
+ *,
1257
+ keepdims: Literal[False] | _NoValueType = ...,
1258
+ ) -> intp: ...
1259
+ @overload
1260
+ def argmin(
1261
+ self: ArrayLike,
1262
+ axis: SupportsIndex | None = None,
1263
+ fill_value: _ScalarLike_co | None = None,
1264
+ out: None = None,
1265
+ *,
1266
+ keepdims: bool | _NoValueType = ...,
1267
+ ) -> Any: ...
1268
+ @overload
1269
+ def argmin(
1270
+ self: ArrayLike,
1271
+ axis: SupportsIndex | None = None,
1272
+ fill_value: _ScalarLike_co | None = None,
1273
+ *,
1274
+ out: _ArrayT,
1275
+ keepdims: bool | _NoValueType = ...,
1276
+ ) -> _ArrayT: ...
1277
+ @overload
1278
+ def argmin(
1279
+ self: ArrayLike,
1280
+ axis: SupportsIndex | None,
1281
+ fill_value: _ScalarLike_co | None,
1282
+ out: _ArrayT,
1283
+ *,
1284
+ keepdims: bool | _NoValueType = ...,
1285
+ ) -> _ArrayT: ...
1286
+
1287
+ #
1288
+ @overload
1289
+ def argmax(
1290
+ self: ArrayLike,
1291
+ axis: None = None,
1292
+ fill_value: _ScalarLike_co | None = None,
1293
+ out: None = None,
1294
+ *,
1295
+ keepdims: Literal[False] | _NoValueType = ...,
1296
+ ) -> intp: ...
1297
+ @overload
1298
+ def argmax(
1299
+ self: ArrayLike,
1300
+ axis: SupportsIndex | None = None,
1301
+ fill_value: _ScalarLike_co | None = None,
1302
+ out: None = None,
1303
+ *,
1304
+ keepdims: bool | _NoValueType = ...,
1305
+ ) -> Any: ...
1306
+ @overload
1307
+ def argmax(
1308
+ self: ArrayLike,
1309
+ axis: SupportsIndex | None = None,
1310
+ fill_value: _ScalarLike_co | None = None,
1311
+ *,
1312
+ out: _ArrayT,
1313
+ keepdims: bool | _NoValueType = ...,
1314
+ ) -> _ArrayT: ...
1315
+ @overload
1316
+ def argmax(
1317
+ self: ArrayLike,
1318
+ axis: SupportsIndex | None,
1319
+ fill_value: _ScalarLike_co | None,
1320
+ out: _ArrayT,
1321
+ *,
1322
+ keepdims: bool | _NoValueType = ...,
1323
+ ) -> _ArrayT: ...
1324
+
1325
+ minimum: _extrema_operation
1326
+ maximum: _extrema_operation
1327
+
1328
+ @overload
1329
+ def take(
1330
+ a: _ArrayLike[_ScalarT],
1331
+ indices: _IntLike_co,
1332
+ axis: None = None,
1333
+ out: None = None,
1334
+ mode: _ModeKind = 'raise'
1335
+ ) -> _ScalarT: ...
1336
+ @overload
1337
+ def take(
1338
+ a: _ArrayLike[_ScalarT],
1339
+ indices: _ArrayLikeInt_co,
1340
+ axis: SupportsIndex | None = None,
1341
+ out: None = None,
1342
+ mode: _ModeKind = 'raise',
1343
+ ) -> _MaskedArray[_ScalarT]: ...
1344
+ @overload
1345
+ def take(
1346
+ a: ArrayLike,
1347
+ indices: _IntLike_co,
1348
+ axis: SupportsIndex | None = None,
1349
+ out: None = None,
1350
+ mode: _ModeKind = 'raise',
1351
+ ) -> Any: ...
1352
+ @overload
1353
+ def take(
1354
+ a: ArrayLike,
1355
+ indices: _ArrayLikeInt_co,
1356
+ axis: SupportsIndex | None = None,
1357
+ out: None = None,
1358
+ mode: _ModeKind = 'raise',
1359
+ ) -> _MaskedArray[Any]: ...
1360
+ @overload
1361
+ def take(
1362
+ a: ArrayLike,
1363
+ indices: _ArrayLikeInt_co,
1364
+ axis: SupportsIndex | None,
1365
+ out: _ArrayT,
1366
+ mode: _ModeKind = 'raise',
1367
+ ) -> _ArrayT: ...
1368
+ @overload
1369
+ def take(
1370
+ a: ArrayLike,
1371
+ indices: _ArrayLikeInt_co,
1372
+ axis: SupportsIndex | None = None,
1373
+ *,
1374
+ out: _ArrayT,
1375
+ mode: _ModeKind = 'raise',
1376
+ ) -> _ArrayT: ...
1377
+
1378
+ def power(a, b, third=...): ...
1379
+ def argsort(a, axis=..., kind=..., order=..., endwith=..., fill_value=..., *, stable=...): ...
1380
+ @overload
1381
+ def sort(
1382
+ a: _ArrayT,
1383
+ axis: SupportsIndex = -1,
1384
+ kind: _SortKind | None = None,
1385
+ order: str | Sequence[str] | None = None,
1386
+ endwith: bool | None = True,
1387
+ fill_value: _ScalarLike_co | None = None,
1388
+ *,
1389
+ stable: Literal[False] | None = False,
1390
+ ) -> _ArrayT: ...
1391
+ @overload
1392
+ def sort(
1393
+ a: ArrayLike,
1394
+ axis: SupportsIndex = -1,
1395
+ kind: _SortKind | None = None,
1396
+ order: str | Sequence[str] | None = None,
1397
+ endwith: bool | None = True,
1398
+ fill_value: _ScalarLike_co | None = None,
1399
+ *,
1400
+ stable: Literal[False] | None = False,
1401
+ ) -> NDArray[Any]: ...
1402
+ @overload
1403
+ def compressed(x: _ArrayLike[_ScalarT_co]) -> _Array1D[_ScalarT_co]: ...
1404
+ @overload
1405
+ def compressed(x: ArrayLike) -> _Array1D[Any]: ...
1406
+ def concatenate(arrays, axis=...): ...
1407
+ def diag(v, k=...): ...
1408
+ def left_shift(a, n): ...
1409
+ def right_shift(a, n): ...
1410
+ def put(a: NDArray[Any], indices: _ArrayLikeInt_co, values: ArrayLike, mode: _ModeKind = 'raise') -> None: ...
1411
+ def putmask(a: NDArray[Any], mask: _ArrayLikeBool_co, values: ArrayLike) -> None: ...
1412
+ def transpose(a, axes=...): ...
1413
+ def reshape(a, new_shape, order=...): ...
1414
+ def resize(x, new_shape): ...
1415
+ def ndim(obj: ArrayLike) -> int: ...
1416
+ def shape(obj): ...
1417
+ def size(obj: ArrayLike, axis: SupportsIndex | None = None) -> int: ...
1418
+ def diff(a, /, n=..., axis=..., prepend=..., append=...): ...
1419
+ def where(condition, x=..., y=...): ...
1420
+ def choose(indices, choices, out=..., mode=...): ...
1421
+ def round_(a, decimals=..., out=...): ...
1422
+ round = round_
1423
+
1424
+ def inner(a, b): ...
1425
+ innerproduct = inner
1426
+
1427
+ def outer(a, b): ...
1428
+ outerproduct = outer
1429
+
1430
+ def correlate(a, v, mode=..., propagate_mask=...): ...
1431
+ def convolve(a, v, mode=..., propagate_mask=...): ...
1432
+
1433
+ def allequal(a: ArrayLike, b: ArrayLike, fill_value: bool = True) -> bool: ...
1434
+
1435
+ def allclose(a: ArrayLike, b: ArrayLike, masked_equal: bool = True, rtol: float = 1e-5, atol: float = 1e-8) -> bool: ...
1436
+
1437
+ def asarray(a, dtype=..., order=...): ...
1438
+ def asanyarray(a, dtype=...): ...
1439
+ def fromflex(fxarray): ...
1440
+
1441
+ class _convert2ma:
1442
+ def __init__(self, /, funcname: str, np_ret: str, np_ma_ret: str, params: dict[str, Any] | None = None) -> None: ...
1443
+ def __call__(self, /, *args: object, **params: object) -> Any: ...
1444
+ def getdoc(self, /, np_ret: str, np_ma_ret: str) -> str | None: ...
1445
+
1446
+ arange: _convert2ma
1447
+ clip: _convert2ma
1448
+ empty: _convert2ma
1449
+ empty_like: _convert2ma
1450
+ frombuffer: _convert2ma
1451
+ fromfunction: _convert2ma
1452
+ identity: _convert2ma
1453
+ indices: _convert2ma
1454
+ ones: _convert2ma
1455
+ ones_like: _convert2ma
1456
+ squeeze: _convert2ma
1457
+ zeros: _convert2ma
1458
+ zeros_like: _convert2ma
1459
+
1460
+ def append(a, b, axis=...): ...
1461
+ def dot(a, b, strict=..., out=...): ...
1462
+ def mask_rowcols(a, axis=...): ...