Как объединить два подсписка из двух разных списков в оригинале - PullRequest
3 голосов
/ 10 марта 2019

В обоих моих списках есть подсписки

list1 = [['+', '+', '-'], ['+', '-', '+'], ['-', '+'], ['+', '+', '+']]

list2 = [['x1', 'x2', '2x3'], ['2x1', 'x2', 'x3'], ['x1', 'x3'], ['3x1', '2x2', '4x3']]

Как я могу объединить list1 и list2 на основе одного индекса и сохранить результат в другом списке?

Мой желаемый результат будет:

list1 = [['+x1', '+x2', '-2x3'], ['+2x1', '-x2', '+x3'],['-x1', '+x3'],['+3x1', '+2x2', '+4x3']]

Пока я пытался сделатьэто только для 1 подсписка, но мне не повезло сохранить подсписок как есть

for pair in itertools.product(list1[0], list2[0]):
    matrix3_with_signs.append(''.join(pair))

Ответы [ 2 ]

4 голосов
/ 10 марта 2019

Проверьте с

x=[[m+n for m,n in zip(x,y)]for x , y in zip(list1,list2)]
x
Out[135]: 
[['+x1', '+x2', '-2x3'],
 ['+2x1', '-x2', '+x3'],
 ['-x1', '+x3'],
 ['+3x1', '+2x2', '+4x3']]
0 голосов
/ 10 марта 2019

Вместо вложенного listcomp вы можете использовать функцию map() с оператором concat (или add):

from operator import concat

[list(map(concat, i, j)) for i, j in zip(list1, list2)]
...