235 lines
5.6 KiB
Python
235 lines
5.6 KiB
Python
import os
|
|
from collections.abc import Sequence, Iterable
|
|
from typing import (
|
|
Any,
|
|
TypeVar,
|
|
overload,
|
|
Protocol,
|
|
)
|
|
|
|
from numpy import (
|
|
format_parser as format_parser,
|
|
record as record,
|
|
recarray as recarray,
|
|
dtype,
|
|
generic,
|
|
void,
|
|
_ByteOrder,
|
|
_SupportsBuffer,
|
|
)
|
|
|
|
from numpy._typing import (
|
|
ArrayLike,
|
|
DTypeLike,
|
|
NDArray,
|
|
_ShapeLike,
|
|
_ArrayLikeVoid_co,
|
|
_NestedSequence,
|
|
)
|
|
|
|
_SCT = TypeVar("_SCT", bound=generic)
|
|
|
|
_RecArray = recarray[Any, dtype[_SCT]]
|
|
|
|
class _SupportsReadInto(Protocol):
|
|
def seek(self, offset: int, whence: int, /) -> object: ...
|
|
def tell(self, /) -> int: ...
|
|
def readinto(self, buffer: memoryview, /) -> int: ...
|
|
|
|
__all__: list[str]
|
|
|
|
@overload
|
|
def fromarrays(
|
|
arrayList: Iterable[ArrayLike],
|
|
dtype: DTypeLike = ...,
|
|
shape: None | _ShapeLike = ...,
|
|
formats: None = ...,
|
|
names: None = ...,
|
|
titles: None = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None = ...,
|
|
) -> _RecArray[Any]: ...
|
|
@overload
|
|
def fromarrays(
|
|
arrayList: Iterable[ArrayLike],
|
|
dtype: None = ...,
|
|
shape: None | _ShapeLike = ...,
|
|
*,
|
|
formats: DTypeLike,
|
|
names: None | str | Sequence[str] = ...,
|
|
titles: None | str | Sequence[str] = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None | _ByteOrder = ...,
|
|
) -> _RecArray[record]: ...
|
|
|
|
@overload
|
|
def fromrecords(
|
|
recList: _ArrayLikeVoid_co | tuple[Any, ...] | _NestedSequence[tuple[Any, ...]],
|
|
dtype: DTypeLike = ...,
|
|
shape: None | _ShapeLike = ...,
|
|
formats: None = ...,
|
|
names: None = ...,
|
|
titles: None = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None = ...,
|
|
) -> _RecArray[record]: ...
|
|
@overload
|
|
def fromrecords(
|
|
recList: _ArrayLikeVoid_co | tuple[Any, ...] | _NestedSequence[tuple[Any, ...]],
|
|
dtype: None = ...,
|
|
shape: None | _ShapeLike = ...,
|
|
*,
|
|
formats: DTypeLike,
|
|
names: None | str | Sequence[str] = ...,
|
|
titles: None | str | Sequence[str] = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None | _ByteOrder = ...,
|
|
) -> _RecArray[record]: ...
|
|
|
|
@overload
|
|
def fromstring(
|
|
datastring: _SupportsBuffer,
|
|
dtype: DTypeLike,
|
|
shape: None | _ShapeLike = ...,
|
|
offset: int = ...,
|
|
formats: None = ...,
|
|
names: None = ...,
|
|
titles: None = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None = ...,
|
|
) -> _RecArray[record]: ...
|
|
@overload
|
|
def fromstring(
|
|
datastring: _SupportsBuffer,
|
|
dtype: None = ...,
|
|
shape: None | _ShapeLike = ...,
|
|
offset: int = ...,
|
|
*,
|
|
formats: DTypeLike,
|
|
names: None | str | Sequence[str] = ...,
|
|
titles: None | str | Sequence[str] = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None | _ByteOrder = ...,
|
|
) -> _RecArray[record]: ...
|
|
|
|
@overload
|
|
def fromfile(
|
|
fd: str | bytes | os.PathLike[str] | os.PathLike[bytes] | _SupportsReadInto,
|
|
dtype: DTypeLike,
|
|
shape: None | _ShapeLike = ...,
|
|
offset: int = ...,
|
|
formats: None = ...,
|
|
names: None = ...,
|
|
titles: None = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None = ...,
|
|
) -> _RecArray[Any]: ...
|
|
@overload
|
|
def fromfile(
|
|
fd: str | bytes | os.PathLike[str] | os.PathLike[bytes] | _SupportsReadInto,
|
|
dtype: None = ...,
|
|
shape: None | _ShapeLike = ...,
|
|
offset: int = ...,
|
|
*,
|
|
formats: DTypeLike,
|
|
names: None | str | Sequence[str] = ...,
|
|
titles: None | str | Sequence[str] = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None | _ByteOrder = ...,
|
|
) -> _RecArray[record]: ...
|
|
|
|
@overload
|
|
def array(
|
|
obj: _SCT | NDArray[_SCT],
|
|
dtype: None = ...,
|
|
shape: None | _ShapeLike = ...,
|
|
offset: int = ...,
|
|
formats: None = ...,
|
|
names: None = ...,
|
|
titles: None = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None = ...,
|
|
copy: bool = ...,
|
|
) -> _RecArray[_SCT]: ...
|
|
@overload
|
|
def array(
|
|
obj: ArrayLike,
|
|
dtype: DTypeLike,
|
|
shape: None | _ShapeLike = ...,
|
|
offset: int = ...,
|
|
formats: None = ...,
|
|
names: None = ...,
|
|
titles: None = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None = ...,
|
|
copy: bool = ...,
|
|
) -> _RecArray[Any]: ...
|
|
@overload
|
|
def array(
|
|
obj: ArrayLike,
|
|
dtype: None = ...,
|
|
shape: None | _ShapeLike = ...,
|
|
offset: int = ...,
|
|
*,
|
|
formats: DTypeLike,
|
|
names: None | str | Sequence[str] = ...,
|
|
titles: None | str | Sequence[str] = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None | _ByteOrder = ...,
|
|
copy: bool = ...,
|
|
) -> _RecArray[record]: ...
|
|
@overload
|
|
def array(
|
|
obj: None,
|
|
dtype: DTypeLike,
|
|
shape: _ShapeLike,
|
|
offset: int = ...,
|
|
formats: None = ...,
|
|
names: None = ...,
|
|
titles: None = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None = ...,
|
|
copy: bool = ...,
|
|
) -> _RecArray[Any]: ...
|
|
@overload
|
|
def array(
|
|
obj: None,
|
|
dtype: None = ...,
|
|
*,
|
|
shape: _ShapeLike,
|
|
offset: int = ...,
|
|
formats: DTypeLike,
|
|
names: None | str | Sequence[str] = ...,
|
|
titles: None | str | Sequence[str] = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None | _ByteOrder = ...,
|
|
copy: bool = ...,
|
|
) -> _RecArray[record]: ...
|
|
@overload
|
|
def array(
|
|
obj: _SupportsReadInto,
|
|
dtype: DTypeLike,
|
|
shape: None | _ShapeLike = ...,
|
|
offset: int = ...,
|
|
formats: None = ...,
|
|
names: None = ...,
|
|
titles: None = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None = ...,
|
|
copy: bool = ...,
|
|
) -> _RecArray[Any]: ...
|
|
@overload
|
|
def array(
|
|
obj: _SupportsReadInto,
|
|
dtype: None = ...,
|
|
shape: None | _ShapeLike = ...,
|
|
offset: int = ...,
|
|
*,
|
|
formats: DTypeLike,
|
|
names: None | str | Sequence[str] = ...,
|
|
titles: None | str | Sequence[str] = ...,
|
|
aligned: bool = ...,
|
|
byteorder: None | _ByteOrder = ...,
|
|
copy: bool = ...,
|
|
) -> _RecArray[record]: ...
|