Подсказки типа Numpy в Python (PEP 484)

Question

Я хотел бы добавить подсказки типа к методу, который принимает в качестве входного значения массив numpy и возвращает строку.Этот массивный массив содержит числа с плавающей точкой, поэтому я попытался:

import numpy as np
def foo(array: np.ndarray[np.float64]) -> str:

Но он не будет работать из-за TypeError: 'type' object is not subscriptable.

Я нашел это , но не смог следоватьобсуждение!

Answer 1

Проверить nptyping .Он предлагает подсказки типов для массивов с номерами.

В вашем случае вы получите:

import numpy as np
from nptyping import Array

def foo(array: Array[np.float64]) -> str:
    ...

Вы также можете проверить свои экземпляры:

arr = np.array([[1.0, 2.0],
                [3.0, 4.0],
                [5.0, 6.0]])
isinstance(arr, Array[np.float64, 3, 2])  # True