我有两个形式的向量
a = [[1,2,3],[1,2,3],[1,2,3]]
b = [[5,6,7],[5,6,7],[5,6,7]]
我想要的输出是
c = [[1,2,3,5,6,7],[1,2,3,5,6,7],[1,2,3,5,6,7]]
我得到了这条线
c = [[a[i],b[i]] for i in range(len(a))]
但我的输出是
[[[1, 2, 3], [5, 6, 7]], [[1, 2, 3], [5, 6, 7]], [[1, 2, 3], [5, 6, 7]]
zip并连接每个配对:
a = [[1,2,3],[1,2,3],[1,2,3]]
b = [[5,6,7],[5,6,7],[5,6,7]]
print([i + j for i,j in zip(a, b)])
这会给你:
[[1, 2, 3, 5, 6, 7], [1, 2, 3, 5, 6, 7], [1, 2, 3, 5, 6, 7]]
或者使用你自己的逻辑:
[a[i] + b[i] for i in range(len(a))]
用 + 连接是关键。如果你要建立索引,我会使用 enumerate :
[ele + b[i] for i, ele in enumerate(a)]