_typing.py

from pathlib import Path
from typing import IO, AnyStr, Iterable, TypeVar, Union

import numpy as np

from pandas._libs import Timestamp
from pandas._libs.tslibs.period import Period
from pandas._libs.tslibs.timedeltas import Timedelta

from pandas.core.dtypes.dtypes import ExtensionDtype
from pandas.core.dtypes.generic import (
    ABCDataFrame,
    ABCExtensionArray,
    ABCIndexClass,
    ABCSeries,
    ABCSparseSeries,
)

AnyArrayLike = TypeVar(
    "AnyArrayLike",
    ABCExtensionArray,
    ABCIndexClass,
    ABCSeries,
    ABCSparseSeries,
    np.ndarray,
)
ArrayLike = TypeVar("ArrayLike", ABCExtensionArray, np.ndarray)
DatetimeLikeScalar = TypeVar("DatetimeLikeScalar", Period, Timestamp, Timedelta)
Dtype = Union[str, np.dtype, ExtensionDtype]
FilePathOrBuffer = Union[str, Path, IO[AnyStr]]

# Type for index and columns of DataFrame
Axes = Iterable[Union[ABCIndexClass, Iterable[str]]]

# Type for data of DataFrame
Data = TypeVar("Data", ABCDataFrame, dict, np.ndarray)

FrameOrSeries = TypeVar("FrameOrSeries", ABCSeries, ABCDataFrame)
Scalar = Union[str, int, float]