Как набрать подсказку np.nan? ⇐ Python

[Anonymous]

У меня есть функция, которая вычисляет значение «статус» машины - либо на (либо на (

Код: Выделить всё

1

) или выключение (

Код: Выделить всё

0

). Но я также хочу, чтобы он вернул np.nan , если входное значение не является или nan (например, что сделает Pandas. Моя цель - применить это на P.DataFrame) . < /p>

Код: Выделить всё

def calc_status(raw_value: int) -> int:
if raw_value is None or np.isnan(raw_value):
return np.nan
# do other checks & calcs
return status

Поскольку функция вернет только 1/0 целых числа, подсказка типа возврата может быть буквальным [0, 1] .
Но функция также может вернуть NAN ... type (np.nan) is float . Я не хочу намекать на Union [Literal [0, 1], float] , потому что этот «поплавок» кажется слишком общим. Функция не может просто вернуть любые поплавки , например, 1,5 или 1,0. Он может только вернуть NAN, который оказывается «float». Code>, но это кажется недействительным, с статическим контролем статического типа Pycharm, показывающей это предупреждение: < /p>

'Literal' может быть параметризован с буквальными INT, Byte и Unicode Строки, лоп, значения перечисления, нет, другие буквальные типы или псевдонимы типов к другим буквальным типам


Подробнее здесь: https://stackoverflow.com/questions/732 ... int-np-nan