我想将一些函数应用于数组数组中的某些元素。
def func_1(num):
return num+1
def func_2(num):
return num+2
test_array = [ [10,1],[10,1],[11,2]]
我想将函数 func_1 应用于每个数组的第一个元素,并将函数 func_2 应用于每个数组的第二个元素。
结果数组将如下所示;
result_array = [ [11,3],[11,3],[12,4]]
我正在使用 python 3.7
请您参考如下方法:
你可以使用 map():
def func_1(num):
return num+1
def func_2(num):
return num+2
test_array = [ [10,1],[10,1],[11,2]]
out = list(map(lambda x: [func_1(x[0]), func_2(x[1])], test_array))
print(out)
打印:
[[11, 3], [11, 3], [12, 4]]
或者使用理解:
out = [[func_1(x), func_2(y)] for x, y in test_array]
