Тип подсказка для NumPy ndarray dtype?

Question

Я бы хотел, чтобы функция включала подсказку типа для NumPy ndarray вместе с его dtype.

Например, со списками можно сделать следующее ...

def foo(bar: List[int]):
   ...

... чтобы дать подсказку типа, что bar должен быть list, состоящим из int.

К сожалению, этот синтаксис вызывает исключения для NumPy ndarray:

def foo(bar: np.ndarray[np.bool]):
   ...

> np.ndarray[np.bool]) (...) TypeError: 'type' object is not subscriptable

Можно ли дать dtype специфичные для типа подсказки для np.ndarray?

Answer 1

Вы можете проверить nptyping :

from nptyping import Array

def foo(bar: Array[np.bool]):
   ...

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

def foo(bar: 'np.ndarray[np.bool]'):
   ...

Answer 2

Насколько я знаю, пока невозможно указать dtype в подсказках типа numpy array в сигнатурах функций.Это планируется реализовать в какой-то момент в будущем.См. NumPy GitHub выпуск # 7370 и NUMPY-заглушки GitHub для получения дополнительной информации о текущем состоянии разработки.