python中两个列表相减的方法

Question

我不知道如何在 python 中创建一个可以计算这个的函数：

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

结果应该是一个从 List1 中减去 List2 的新列表，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

此解决方案使用 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 3

您可以按照@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 函数一样，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]

编辑：正如下面 @Cat 所建议的，最好使用 带有imap()的operator.sub()函数：

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

Answer 4

下面还有另一个解决方案：

>>> 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 参考map 你会发现你可以将多个迭代传递给 map

Answer 5

您可以按照以下方式

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

输出 [0, -2, 4]