add all arrays

Author: cmp0xff

pandas-stubs/_libs/arrays.pyi

@@ -1 +1,38 @@
-class NDArrayBacked: ...
+from collections.abc import Sequence
+from typing import Any
+
+import numpy as np
+from typing_extensions import Self
+
+from pandas._typing import (
+    AnyArrayLikeInt,
+    AxisInt,
+    DtypeObj,
+    Shape,
+)
+
+class NDArrayBacked:
+    _dtype: DtypeObj
+    _ndarray: np.ndarray
+    def __setstate__(self, state: Any) -> None: ...
+    def __len__(self) -> int: ...
+    @property
+    def shape(self) -> Shape: ...
+    @property
+    def ndim(self) -> int: ...
+    @property
+    def size(self) -> int: ...
+    @property
+    def nbytes(self) -> int: ...
+    def copy(self, order=...) -> Self: ...
+    def delete(self, loc, axis=...) -> Self: ...
+    def swapaxes(self, axis1, axis2) -> Self: ...
+    def repeat(
+        self,
+        repeats: int | Sequence[int] | AnyArrayLikeInt,
+        axis: AxisInt | None = None,
+    ) -> Self: ...
+    def reshape(self, *args: Any, **kwargs: Any) -> Self: ...
+    def ravel(self, order=...) -> Self: ...
+    @property
+    def T(self) -> Self: ...

pandas-stubs/core/arrays/__init__.pyi

@@ -1,15 +1,43 @@
+from pandas.core.arrays.arrow import ArrowExtensionArray
 from pandas.core.arrays.base import (
-    ExtensionArray as ExtensionArray,
-    ExtensionOpsMixin as ExtensionOpsMixin,
-    ExtensionScalarOpsMixin as ExtensionScalarOpsMixin,
+    ExtensionArray,
+    ExtensionOpsMixin,
+    ExtensionScalarOpsMixin,
 )
-from pandas.core.arrays.boolean import BooleanArray as BooleanArray
-from pandas.core.arrays.categorical import Categorical as Categorical
-from pandas.core.arrays.datetimes import DatetimeArray as DatetimeArray
-from pandas.core.arrays.integer import IntegerArray as IntegerArray
-from pandas.core.arrays.interval import IntervalArray as IntervalArray
-from pandas.core.arrays.numpy_ import PandasArray as PandasArray
-from pandas.core.arrays.period import PeriodArray as PeriodArray
-from pandas.core.arrays.sparse import SparseArray as SparseArray
-from pandas.core.arrays.string_ import StringArray as StringArray
-from pandas.core.arrays.timedeltas import TimedeltaArray as TimedeltaArray
+from pandas.core.arrays.boolean import BooleanArray
+from pandas.core.arrays.categorical import Categorical
+from pandas.core.arrays.datetimes import DatetimeArray
+from pandas.core.arrays.floating import FloatingArray
+from pandas.core.arrays.integer import IntegerArray
+from pandas.core.arrays.interval import IntervalArray
+from pandas.core.arrays.masked import BaseMaskedArray
+from pandas.core.arrays.numpy_ import NumpyExtensionArray
+from pandas.core.arrays.period import (
+    PeriodArray,
+    period_array,
+)
+from pandas.core.arrays.sparse import SparseArray
+from pandas.core.arrays.string_ import StringArray
+from pandas.core.arrays.string_arrow import ArrowStringArray
+from pandas.core.arrays.timedeltas import TimedeltaArray
+
+__all__ = [
+    "ArrowExtensionArray",
+    "ArrowStringArray",
+    "BaseMaskedArray",
+    "BooleanArray",
+    "Categorical",
+    "DatetimeArray",
+    "ExtensionArray",
+    "ExtensionOpsMixin",
+    "ExtensionScalarOpsMixin",
+    "FloatingArray",
+    "IntegerArray",
+    "IntervalArray",
+    "NumpyExtensionArray",
+    "PeriodArray",
+    "SparseArray",
+    "StringArray",
+    "TimedeltaArray",
+    "period_array",
+]

pandas-stubs/core/arrays/_arrow_string_mixins.pyi

@@ -0,0 +1 @@
+class ArrowStringArrayMixin: ...

pandas-stubs/core/arrays/_mixins.pyi

@@ -1,5 +1,15 @@
 from pandas.core.arrays.base import ExtensionArray
+from pandas.core.series import Series
+from typing_extensions import Self
 
 from pandas._libs.arrays import NDArrayBacked
+from pandas._typing import (
+    AxisInt,
+    Scalar,
+)
 
-class NDArrayBackedExtensionArray(NDArrayBacked, ExtensionArray): ...
+class NDArrayBackedExtensionArray(NDArrayBacked, ExtensionArray):
+    def argmin(self, axis: AxisInt = 0, skipna: bool = True) -> int: ...
+    def argmax(self, axis: AxisInt = 0, skipna: bool = True) -> int: ...
+    def insert(self, loc: int, item: Scalar) -> Self: ...
+    def value_counts(self, dropna: bool = True) -> Series[int]: ...

pandas-stubs/core/arrays/arrow/__init__.pyi

@@ -0,0 +1,7 @@
+from pandas.core.arrays.arrow.accessors import (
+    ListAccessor,
+    StructAccessor,
+)
+from pandas.core.arrays.arrow.array import ArrowExtensionArray
+
+__all__ = ["ArrowExtensionArray", "ListAccessor", "StructAccessor"]

pandas-stubs/core/arrays/arrow/accessors.pyi

@@ -0,0 +1,5 @@
+from abc import ABCMeta
+
+class ArrowAccessor(metaclass=ABCMeta): ...
+class ListAccessor(ArrowAccessor): ...
+class StructAccessor(ArrowAccessor): ...

pandas-stubs/core/arrays/arrow/array.pyi

@@ -0,0 +1,7 @@
+from pandas.core.arraylike import OpsMixin
+from pandas.core.arrays._arrow_string_mixins import ArrowStringArrayMixin
+from pandas.core.arrays.base import ExtensionArraySupportsAnyAll
+
+class ArrowExtensionArray(
+    OpsMixin, ExtensionArraySupportsAnyAll, ArrowStringArrayMixin
+): ...

pandas-stubs/core/arrays/base.pyi

@@ -87,7 +87,7 @@ class ExtensionArray:
     ) -> np.intp: ...
     def factorize(self, use_na_sentinel: bool = True) -> tuple[np_1darray, Self]: ...
     def repeat(
-        self, repeats: int | AnyArrayLikeInt | Sequence[int], axis: None = None
+        self, repeats: int | Sequence[int] | AnyArrayLikeInt, axis: None = None
     ) -> Self: ...
     def take(
         self,
@@ -114,6 +114,8 @@ class ExtensionArray:
         **kwargs: Any,
     ) -> Self: ...
 
+class ExtensionArraySupportsAnyAll(ExtensionArray): ...
+
 class ExtensionOpsMixin:
     @classmethod
     def _add_arithmetic_ops(cls) -> None: ...

pandas-stubs/core/arrays/period.pyi

@@ -1,12 +1,19 @@
+from collections.abc import Sequence
+
 from pandas import PeriodDtype
 from pandas.core.arrays.datetimelike import (
     DatelikeOps,
     DatetimeLikeArrayMixin,
 )
 
 from pandas._libs.tslibs import Timestamp
+from pandas._libs.tslibs.offsets import (
+    BaseOffset,
+    Tick,
+)
 from pandas._libs.tslibs.period import Period
 from pandas._typing import (
+    AnyArrayLike,
     NpDtype,
     PeriodFrequency,
     np_1darray,
@@ -47,3 +54,9 @@ class PeriodArray(DatetimeLikeArrayMixin, DatelikeOps):
         self, freq: PeriodFrequency | None = None, how: str = ...
     ) -> Timestamp: ...
     def asfreq(self, freq: str | None = ..., how: str = "E") -> Period: ...
+
+def period_array(
+    data: Sequence[Period | str | None] | AnyArrayLike,
+    freq: str | Tick | BaseOffset | None = None,
+    copy: bool = False,
+) -> PeriodArray: ...

pandas-stubs/core/arrays/string_.pyi

@@ -1,6 +1,7 @@
 from typing import Literal
 
-from pandas.core.arrays import PandasArray
+from pandas.core.arrays.base import ExtensionArray
+from pandas.core.arrays.numpy_ import NumpyExtensionArray
 
 from pandas._libs.missing import NAType
 
@@ -11,7 +12,9 @@ class StringDtype(ExtensionDtype):
     @property
     def na_value(self) -> NAType: ...
 
-class StringArray(PandasArray):
+class BaseStringArray(ExtensionArray): ...
+
+class StringArray(BaseStringArray, NumpyExtensionArray):
     def __init__(self, values, copy: bool = ...) -> None: ...
     def __arrow_array__(self, type=...): ...
     def __setitem__(self, key, value) -> None: ...