我试图在Python中使用map()和带有if条件的函数组合两个2d数组。
这很有效(但我需要的if条件不满足):
list1 = np.array([np.array([1,2,0]),np.array([1,2,3]),np.array([0,0,3])])
list2 = np.array([np.array([4,5,6]),np.array([4,5,6]),np.array([4,5,6])])
collapsed = list(map(lambda x,y: x + y, list1, list2))
并返回:
collapsed = [array([5, 7, 6]), array([5, 7, 9]), array([4, 5, 9])]
以下返回:
collapsed = list(map(lambda x,y: x + y if x == 0 else x, list1, list2))
我所期望的是:
collapsed = [[1,2,6][1,2,3][4,5,3]]