103 lines
2.8 KiB
Python
103 lines
2.8 KiB
Python
|
from typing import (
|
||
|
|
Iterator,
|
||
|
|
Sequence,
|
||
|
|
final,
|
||
|
|
overload,
|
||
|
|
)
|
||
|
|
import weakref
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
from pandas._typing import (
|
||
|
|
ArrayLike,
|
||
|
|
T,
|
||
|
|
npt,
|
||
|
|
)
|
||
|
|
|
||
|
|
from pandas import Index
|
||
|
|
from pandas.core.arrays._mixins import NDArrayBackedExtensionArray
|
||
|
|
from pandas.core.internals.blocks import Block as B
|
||
|
|
|
||
|
|
def slice_len(slc: slice, objlen: int = ...) -> int: ...
|
||
|
|
def get_blkno_indexers(
|
||
|
|
blknos: np.ndarray, # int64_t[:]
|
||
|
|
group: bool = ...,
|
||
|
|
) -> list[tuple[int, slice | np.ndarray]]: ...
|
||
|
|
def get_blkno_placements(
|
||
|
|
blknos: np.ndarray,
|
||
|
|
group: bool = ...,
|
||
|
|
) -> Iterator[tuple[int, BlockPlacement]]: ...
|
||
|
|
def update_blklocs_and_blknos(
|
||
|
|
blklocs: npt.NDArray[np.intp],
|
||
|
|
blknos: npt.NDArray[np.intp],
|
||
|
|
loc: int,
|
||
|
|
nblocks: int,
|
||
|
|
) -> tuple[npt.NDArray[np.intp], npt.NDArray[np.intp]]: ...
|
||
|
|
@final
|
||
|
|
class BlockPlacement:
|
||
|
|
def __init__(self, val: int | slice | np.ndarray) -> None: ...
|
||
|
|
@property
|
||
|
|
def indexer(self) -> np.ndarray | slice: ...
|
||
|
|
@property
|
||
|
|
def as_array(self) -> np.ndarray: ...
|
||
|
|
@property
|
||
|
|
def as_slice(self) -> slice: ...
|
||
|
|
@property
|
||
|
|
def is_slice_like(self) -> bool: ...
|
||
|
|
@overload
|
||
|
|
def __getitem__(
|
||
|
|
self, loc: slice | Sequence[int] | npt.NDArray[np.intp]
|
||
|
|
) -> BlockPlacement: ...
|
||
|
|
@overload
|
||
|
|
def __getitem__(self, loc: int) -> int: ...
|
||
|
|
def __iter__(self) -> Iterator[int]: ...
|
||
|
|
def __len__(self) -> int: ...
|
||
|
|
def delete(self, loc) -> BlockPlacement: ...
|
||
|
|
def append(self, others: list[BlockPlacement]) -> BlockPlacement: ...
|
||
|
|
def tile_for_unstack(self, factor: int) -> npt.NDArray[np.intp]: ...
|
||
|
|
|
||
|
|
class SharedBlock:
|
||
|
|
_mgr_locs: BlockPlacement
|
||
|
|
ndim: int
|
||
|
|
values: ArrayLike
|
||
|
|
refs: BlockValuesRefs
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
values: ArrayLike,
|
||
|
|
placement: BlockPlacement,
|
||
|
|
ndim: int,
|
||
|
|
refs: BlockValuesRefs | None = ...,
|
||
|
|
) -> None: ...
|
||
|
|
|
||
|
|
class NumpyBlock(SharedBlock):
|
||
|
|
values: np.ndarray
|
||
|
|
@final
|
||
|
|
def getitem_block_index(self: T, slicer: slice) -> T: ...
|
||
|
|
|
||
|
|
class NDArrayBackedBlock(SharedBlock):
|
||
|
|
values: NDArrayBackedExtensionArray
|
||
|
|
@final
|
||
|
|
def getitem_block_index(self: T, slicer: slice) -> T: ...
|
||
|
|
|
||
|
|
class Block(SharedBlock): ...
|
||
|
|
|
||
|
|
class BlockManager:
|
||
|
|
blocks: tuple[B, ...]
|
||
|
|
axes: list[Index]
|
||
|
|
_known_consolidated: bool
|
||
|
|
_is_consolidated: bool
|
||
|
|
_blknos: np.ndarray
|
||
|
|
_blklocs: np.ndarray
|
||
|
|
def __init__(
|
||
|
|
self, blocks: tuple[B, ...], axes: list[Index], verify_integrity=...
|
||
|
|
) -> None: ...
|
||
|
|
def get_slice(self: T, slobj: slice, axis: int = ...) -> T: ...
|
||
|
|
def _rebuild_blknos_and_blklocs(self) -> None: ...
|
||
|
|
|
||
|
|
class BlockValuesRefs:
|
||
|
|
referenced_blocks: list[weakref.ref]
|
||
|
|
def __init__(self, blk: SharedBlock | None = ...) -> None: ...
|
||
|
|
def add_reference(self, blk: SharedBlock) -> None: ...
|
||
|
|
def add_index_reference(self, index: object) -> None: ...
|
||
|
|
def has_reference(self) -> bool: ...
|