python 的 typing hint 可以给 ndarray 注释 shape 吗？

Question

直接 def func() -> ndarray 让人对 shape 和 size 、dtype 这些摸不着头脑

python 的 typing hint 可以在注释 ndarray 的时候，添加 shape 和 size 、dtype 这些信息吗？

Answer 1

不支持用其他方式吧：

import numpy as np
from numpy import ndarray

def my_function(arr: ndarray) -> ndarray:
    """
    Process the input array and return a new array.

    Parameters:
        arr (ndarray): Input array with shape (m, n) and dtype float64.

    Returns:
        ndarray: Output array with shape (n, m) and dtype float64.
    """
    return arr.T

Answer 2

试试这个？https://pypi.org/project/nptyping/