Как вычесть два списка в Python

Question

Я не могу понять, как сделать функцию в Python, который можно вычислить следующим образом:

List1=[3,5,6] 
List2=[3,7,2] 

и результат должен быть новый список, который вычитает List2 из List1, List3=[0,-2,4] ! Я знаю, что мне как-то придется использовать zip-функцию. Делая это, я получаю: ([(3,3), (5,7), (6,2)]), но я не знаю, что теперь делать?

Answer 1

Попробуйте это:

[x1 - x2 for (x1, x2) in zip(List1, List2)] 

Это использует zip, списочные и деструктурирующий.

Answer 2

Вы можете использовать list comprehension, как предложено @Matt. Вы можете также использовать itertools - более конкретно, imap() функцию:

>>> from itertools import imap 
>>> from operator import sub 
>>> a = [3,5,6] 
>>> b = [3,7,2] 
>>> imap(int.__sub__, a, b) 
<itertools.imap object at 0x50e1b0> 
>>> for i in imap(int.__sub__, a, b): 
...  print i 
... 
0 
-2 
4 

Как и все itertools Funcitons, imap() возвращает итератор. Вы можете создать список передать его в качестве параметра для list() конструктора:

>>> list(imap(int.__sub__, a, b)) 
[0, -2, 4] 
>>> list(imap(lambda m, n: m-n, a, b)) # Using lambda 
[0, -2, 4] 

EDIT: Как было предложено @Cat ниже, было бы лучше использовать operator.sub() функцию с imap():

>>> from operator import sub 
>>> list(imap(sub, a, b)) 
[0, -2, 4] 

Answer 3

Еще одно решение ниже:

>>> a = [3,5,6] 
>>> b = [3,7,2] 
>>> list(map(int.__sub__, a, b)) # for python3.x 
[0, -2, 4] 
>>> map(int.__sub__, a, b) # and for python2.x 
[0, -2, 4] 

Сложение: Просто проверьте python reference of map и вы увидите, вы могли бы передать более одного Iterable к map

Answer 4

Это решение использует numpy. Это имеет смысл только для больших списков, так как есть некоторые накладные расходы для создания массивов numpy. OTOH, для чего угодно, кроме коротких списков, это будет невероятно быстро.

>>> import numpy as np 
>>> a = [3,5,6] 
>>> b = [3,7,2] 
>>> list(np.array(a) - np.array(b)) 
[0, -2, 4] 

Answer 5

Вы можете сделать это следующим образом

List1 = [3,5,6] 
List2 = [3,7,2] 
ans = [List1[i]-List2[i] for i in xrange(min(len(List1), len(List2)))] 
print ans 

, который выводит [0, -2, 4]