是否有一种方法来简化这个函数在python中使用单行理解?

Question

简单的问题，正如标题所暗示的那样。我希望使用zip功能，但由于某种原因不能让它工作。

def tuple_sum(A, B):
out = []
for a,b in [x for x in zip(A,B)]:
out1 = []
for a1, b1 in zip(a, b):
out1.append(a1+b1)
out.append(out1)
return out

Answer 1

也许是这样的?

A = [[1, 2], [3, 4]]
B = [[5, 6], [7, 8]]
s = [[a1 + b1 for a1, b1 in zip(a, b)] for a, b in zip(A, B)]
print(s)  # [[6, 8], [10, 12]]