如何传递矢量化 Python 的列表列表

Question

对于python代码的矢量化，我看到了一个传递要执行函数的数字列表的示例。 例：

def myfunc(a, b):
"Return a-b if a>b, otherwise return a+b"
if a > b:
return a - b
else:
return a + b
vfunc = np.vectorize(myfunc)
vfunc([1, 2, 3, 4], 2)

但是我们在函数中的参数是列表，如何进行矢量化。

例：

def myfunc(a, b):
#a is the list
if a[0] > a[1]:
return a[0] - a[1]

Answer 1

因为 a 是列表对象。尝试使用循环方法循环访问列表中的每个值。

def myfunc(a, b):
# loop depends on the length of the list
for i in range(len(a)):
if a[i] > b:
return a[i] - b
else:
return a[i] + b