| from typing import ( |
| Iterator, |
| Sequence, |
| final, |
| overload, |
| ) |
| import weakref |
|
|
| import numpy as np |
|
|
| from pandas._typing import ( |
| ArrayLike, |
| Self, |
| npt, |
| ) |
|
|
| from pandas import Index |
| from pandas.core.internals.blocks import Block as B |
|
|
| def slice_len(slc: slice, objlen: int = ...) -> int: ... |
| def get_concat_blkno_indexers( |
| blknos_list: list[npt.NDArray[np.intp]], |
| ) -> list[tuple[npt.NDArray[np.intp], BlockPlacement]]: ... |
| def get_blkno_indexers( |
| blknos: np.ndarray, |
| 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 add(self, other) -> BlockPlacement: ... |
| def append(self, others: list[BlockPlacement]) -> BlockPlacement: ... |
| def tile_for_unstack(self, factor: int) -> npt.NDArray[np.intp]: ... |
|
|
| class Block: |
| _mgr_locs: BlockPlacement |
| ndim: int |
| values: ArrayLike |
| refs: BlockValuesRefs |
| def __init__( |
| self, |
| values: ArrayLike, |
| placement: BlockPlacement, |
| ndim: int, |
| refs: BlockValuesRefs | None = ..., |
| ) -> None: ... |
| def slice_block_rows(self, slicer: slice) -> Self: ... |
|
|
| 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, slobj: slice, axis: int = ...) -> Self: ... |
| def _rebuild_blknos_and_blklocs(self) -> None: ... |
|
|
| class BlockValuesRefs: |
| referenced_blocks: list[weakref.ref] |
| def __init__(self, blk: Block | None = ...) -> None: ... |
| def add_reference(self, blk: Block) -> None: ... |
| def add_index_reference(self, index: Index) -> None: ... |
| def has_reference(self) -> bool: ... |
|
|