4

我刚刚发现了 python 3 ( https://www.python.org/dev/peps/pep-3107/ ) 的函数注释,这对于记录参数或返回类型来说似乎很棒。它还在我的 pycharm IDE 中提供了更好的智能感知。

我对输入类型模糊的参数有疑问。例如,它可以是一个列表或 numpy 数组或一些“类似数组”的数量。将此类输入参数注释到函数的最佳方法是什么?例子:

import numpy as np

def fun(data: np.ndarray) # can also be a list
    pass

我还有另一种情况,输入可以是两种类型中的任何一种。例子:

def fun(header: Nifti1Header) # can also be Nifti2Header
    pass

记录这些参数输入的最佳方式是什么?

4

2 回答 2

5

如果你使用的是python3.5,最好的方法是使用typing.Union

>>> from typing import Union
>>> import numpy as np
>>> def fun(data: Union[np.ndarray, list]):
        pass

如果您一直使用 Union[t1, t2, ...],您也可以使用 typing.TypeVar。(此外,您可以比代码中的许多联合更轻松地从 TypeVar 添加和删除类型)

>>> from typing import TypeVar
>>> import numpy as np
>>> import array
>>> Ar = TypeVar('Ar', np.ndarray, list, array.array)

然后,此代码会将 Ar 与列表、array.arrays 和 numpy 数组相关联。

于 2016-03-31T13:50:00.903 回答
4

您必须弄清楚您的功能共享的合法输入有哪些共同财产。对于第一个,它看起来应该是可迭代的:

>>> from collections import Iterable
>>> def fun(data: Iterable):
...     pass
... 
>>> isinstance(np.ndarray(0), Iterable)
True
>>> isinstance([], Iterable)
True
>>> isinstance(23, Iterable)
False

对于您的标头,它闻起来像Nifti1Header并且Nifti2Header应该继承自一个公共基类。如果该函数应该适用于任何派生类,为什么不使用基类对其进行注释呢?

于 2016-03-31T13:47:30.367 回答