72

在 Python 2.7 中执行元组算术的最优雅和简洁的方式是什么（没有创建我自己的带有运算符重载的类）？

假设我有两个元组：

a = (10, 10)
b = (4, 4)

我的预期结果是

c = a - b = (6, 6)

我目前使用：

c = (a[0] - b[0], a[1] - b[1])

我也试过：

c = tuple([(i - j) for i in a for j in b])

但结果是(6, 6, 6, 6)。我相信上面的代码可以作为嵌套的 for 循环，导致 4 次迭代和 4 个值。

4

8 回答 8

85

如果您正在寻找快速，您可以使用 numpy：

>>> import numpy
>>> numpy.subtract((10, 10), (4, 4))
array([6, 6])

如果你想把它保存在一个元组中：

>>> tuple(numpy.subtract((10, 10), (4, 4)))
(6, 6)
于 2013-07-02T05:52:43.153 回答
44

一种选择是，

>>> from operator import sub
>>> c = tuple(map(sub, a, b))
>>> c
(6, 6)

并且itertools.imap可以作为替代品map。

当然，您也可以使用operatorto add、mul、div等其他函数。

但我会认真考虑转移到另一种数据结构，因为我认为这类问题不适合tuples

于 2013-07-02T05:44:18.553 回答
35

使用zip和生成器表达式：

c = tuple(x-y for x, y in zip(a, b))

演示：

>>> a = (10, 10)
>>> b = (4, 4)
>>> c = tuple(x-y for x, y in zip(a, b))
>>> c
(6, 6)

用于itertools.izip内存高效的解决方案。

帮助zip：

>>> print zip.__doc__
zip(seq1 [, seq2 [...]]) -> [(seq1[0], seq2[0] ...), (...)]

Return a list of tuples, where each tuple contains the i-th element
from each of the argument sequences.  The returned list is truncated
in length to the length of the shortest argument sequence.
于 2013-07-02T05:41:05.243 回答
10

JFYI，在我的笔记本电脑中执行 100000 次迭代的时间

np.subtract(a, b): 0.18578505516052246

tuple(x - y for x, y in zip(a, b)): 0.09348797798156738

tuple(map(lambda x, y: x - y, a, b)): 0.07900381088256836

from operator import sub tuple(map(sub, a, b)): 0.044342041015625

运算符对我来说看起来更优雅。

于 2018-11-20T02:24:51.873 回答
7

这也可以在没有导入的情况下很好地完成，尽管 lambda 通常是不可取的：

tuple(map(lambda x, y: x - y, a, b))

如果您要获取二维坐标平面上两点之间的距离，则应使用对的减法的绝对值。

tuple(map(lambda x ,y: abs(x - y), a, b))
于 2016-08-04T08:47:36.057 回答
1

作为 Kohei Kawasaki 答案的补充，对于速度，原始解决方案实际上是最快的（至少对于长度为两个元组）。

>>> timeit.timeit('tuple(map(add, a, b))',number=1000000,setup='from operator import add; a=(10,11); b=(1,2)')
0.6502681339999867
>>> timeit.timeit('(a[0] - b[0], a[1] - b[1])',number=1000000,setup='from operator import add; a=(10,11); b=(1,2)')
0.19015854899998885
>>>
于 2019-08-09T15:51:34.327 回答
1

由于在您的问题中只有 2 数元组的示例，因此对于此类类似坐标的元组，您可能会使用简单的内置“复杂”类：

>>> a=complex(7,5)
>>> b=complex(1,2)
>>> a-b
>>> c=a-b
>>> c
(6+3j)
>>> c.real
6.0
>>> c.imag
3.0
于 2020-03-06T07:00:27.347 回答
0

我的元素元组算术助手

支持的操作：+、-、/、*、d

operation = 'd' 计算二维坐标平面上两点之间的距离

def tuplengine(tuple1, tuple2, operation):
    """ 
    quick and dirty, element-wise, tuple arithmetic helper,
    created on Sun May 28 07:06:16 2017
    ...
    tuple1, tuple2: [named]tuples, both same length
    operation: '+', '-', '/', '*', 'd'
    operation 'd' returns distance between two points on a 2D coordinate plane (absolute value of the subtraction of pairs)
    """
    assert len(tuple1) == len(tuple2), "tuple sizes doesn't match, tuple1: {}, tuple2: {}".format(len(tuple1), len(tuple2))
    assert isinstance(tuple1, tuple) or tuple in type(tuple1).__bases__, "tuple1: not a [named]tuple"
    assert isinstance(tuple2, tuple) or tuple in type(tuple2).__bases__, "tuple2: not a [named]tuple"
    assert operation in list("+-/*d"), "operation has to be one of ['+','-','/','*','d']"
    return eval("tuple( a{}b for a, b in zip( tuple1, tuple2 ))".format(operation)) \
    if not operation == "d" \
      else eval("tuple( abs(a-b) for a, b in zip( tuple1, tuple2 ))")
于 2017-05-28T07:48:58.520 回答