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

• Z时代
• 2024-02-24
• 分类：IT

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

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

回答：

不支持用其他方式吧：

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

回答：

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

以上是 python 的 typing hint 可以给 ndarray 注释 shape 吗？ 的全部内容， 来源链接： utcz.com/p/938982.html