Есть ли способ объединить такой список?

Question

предположим, что a и b являются списком.

a = [[1], [2]]   
b = [[5, 6, 7], [3, 4, 5]]  

Я хочу получить список, который

[[1,5,6,7], [2,3,4,5]]  

Есть ли способ сделать это эффективно?Все списки или массивы в порядке.

Answer 1

zip ваш друг:

>>> a = [[1], [2]]
>>> b = [[5, 6, 7], [3, 4, 5]]
>>> [x+y for x, y in zip(a, b)]
[[1, 5, 6, 7], [2, 3, 4, 5]]

Вы также можете использовать map; модуль operator предоставляет готовое определение lambda x,y: x + y для таких применений.

>>> import operator
>>> list(map(operator.add, a, b))