我有两个n长度的元组,我需要检查相同位置的所有元素是否相同,除了位置w中的元素。这就是我写的:
if all(tup1[i] == tup2[i] for i in xrange(n) if i != w):
...
为了避免循环(因为这段代码将被多次使用),我尝试使用切片。不幸的是,这不起作用:
if tup1[w-1:w-n:-1] == tup2[w-1:w-n:-1]:
...
我有责任写这样的东西吗?
if tup1[:w-1] == tup2[:w-1] and tup1[w+1:] == tup2[w+1:]
是不是有更优雅的方法?
或者循环和切片都不好,有更好的方法来获得我正在寻找的结果? (我不能使用过滤器,因为可能存在与位置w中的元素具有相同值的元素)
答案 0 :(得分:0)
我认为您已经找到了最佳解决方案:
tup1[:w-1] == tup2[:w-1] and tup1[w+1:] == tup2[w+1:]
如果元组非常长并且您不想复制数据并且您想要早期行为,那么使用 itertools 和运算符会有一个更复杂的替代方法:
>>> from operator import eq
>>> from itertools import imap
>>> w = 5
>>> t1 = (10, 20, 30, 40, -1, 50, 60, 70)
>>> t2 = (10, 20, 30, 40, -1, 50, 60, 70)
>>> it1, it2 = iter(t1), iter(t2)
>>> all(imap(eq, islice(it1, w-1), islice(it2, w-1))) \
and (next(it1, True) or True) and (next(it2, True) and True) \
and all(imap(eq, it1, it2))
True
这是很多设置工作,并且逐步没有像元组切片一样快,但它确实避免了复制所有数据,并确实提前了。
在非极端情况下,我会坚持使用双切片 - 元组 - 等式解决方案。