python - 获得所需输出的替代方法 "Numpy Vectorized "

Question

是否有替代方法来获得所需的输出？现在通过调用 numpy.vectorize 函数导入 numpy 包，它工作正常。

我不想导入包，我想通过编写原始 python 代码来实现它。有什么建议吗？？

def comp(a, b):
    if a < b:
        return a+b
    else:
        return a-b

import numpy as np
b = np.vectorize(comp)
c = b([1, 2, 3], 2)
c

应该得到像 [3, 0, 1] 这样的结果

Answer 1

只需使用列表理解:

[comp(x, 2) for x in [1,2,3]]

或者映射两个列表上的函数:

list(map(comp, [1,2,3], [2,2,2]))

或者使用 functools.partial 并将其结果映射到列表中:

import functools
list(map(functools.partial(comp, b=2), [1,2,3]))