python - 标准化的 numpy/scipy `asarray` 装饰器

Question

是否有用于在函数参数上应用 asarray 的标准化装饰器？

即我可以写的东西:

@array_args  # this decorator automatically calls asarray on arguments
def f(x,y):
     return x/y

代替

def f(x,y):
    (x,y)=(numpy.asarray(x), numpy.asarray(y))
     return x/y

我只找到了这个 discussion ，所以我可以把装饰器拉到那里；但我会更喜欢使用已集成到 numpy 中的版本。

Answer 1

据我所知它不存在，但是写一个很简单:

from functools import wraps


def array_args(func):
    @wraps(func)
    def wrapper(*args):
        arrays = map(np.asarray, args)
        return func(*arrays)

    return wrapper