我有一个函数scalar_func(*args)
,它接受一个可变数标量数。它对它们进行一些数学运算并输出一个标量。作为一个简单的例子,我们假设scalar_func
将每个数字相乘:
def scalar_func(*args):
out = 1
for arg in args:
out *= arg
return out
我想scalar_func
处理列表。为此,我制作了另一个函数list_func(*args)
。它接受可变数量的列表并创建一个新列表,如下所示:
def list_func(*args):
out = []
for i in range(len(arg[0])):
out.append(scalar_func(arg[0][i], arg[1][i], arg[2][i]...)
return out
显然,这个函数只是伪代码。我该如何实施list_func
?