类似数组的类型提示

Question

对于接受一维数组对象的函数，正确的类型提示是什么？更具体地说，我的函数使用np.percentile并且我希望np.percentile在它接受的数组类型（List、pandas Series、numpy array 等）方面“匹配”它的灵活性。下面说明了我正在寻找的内容：

def foo(arr: array-like) -> float:
    p = np.percentile(arr, 50)
    return p

Answer 1

使用numpy.typing.ArrayLike：

from numpy.typing import ArrayLike

def foo(arr: ArrayLike) -> float:
    p = np.percentile(arr, 50)
    return p