在列表列表中添加对中的项目

时间:2015-02-08 14:29:12

标签: python list python-3.x

如何在列表列表中压缩对?

 A=[[ 1,2 ],[ 3  ,  4]]
 B=[[ 4,5 ],[ 8  ,  9]]
 ->(1,4),(2,5),(3,8),(4,9)

我尝试了zip(*A,*B)之类的内容,但我得到SyntaxError: only named arguments may follow *expression

最后我要做的就是添加它们:

 A=[[ 1,2 ],[ 3  ,  4]]
 B=[[ 4,5 ],[ 8  ,  9]]
  =[[ 5,7 ],[ 11 , 13]]

(也不起作用):

add= [i+j for i,j in zip(*A,*B)]

2 个答案:

答案 0 :(得分:2)

首先,zip AB,然后是zip上一个zip给出的列表,就像这样

result = []
for items in zip(A, B):
    for data in zip(*items):
        result.append(data)

同样可以简洁地写成列表理解,就像这样

>>> [data for items in zip(A, B) for data in zip(*items)]
[(1, 4), (2, 5), (3, 8), (4, 9)]

答案 1 :(得分:0)

考虑使用numpy

>>> A = [[1, 2], [3, 4]]
>>> B = [[4, 5], [8, 9]]
>>> import numpy
>>> numpy.array(A) + numpy.array(B)
array([[ 5,  7],
       [11, 13]])

>>> list(map(list, _))
[[5, 7], [11, 13]]