检查两台发电机的不同之处

Question

为了简单起见，假设我有两个生成器：

it1 = iter([1,2,3,4])
it2 = iter([1,2,10,20])

知道第一个差值（位置2）在哪里或者它们是否相等的最佳方法是什么？我想避免显式循环：

for pos, v1, v2 in enumerate(izip(it1, it2)):
   if v1 != v2: return pos
return None

并使用功能编程：

try:
  return next(pos for pos, (v1, v2) in enumerate(izip(it1, it2)) if v1 != v2)
except StopIteration:
  return None

重点是我使用的解决方案很难看，比显式循环更冗长。。。

你能做得更好吗？

Answer 1

您可以通过给next:一个默认值来简化第二个示例

return next((pos for pos, (v1, v2) in enumerate(izip(it1, it2)) if v1 != v2), None)

来自文档：

next(iterator[, default])

通过调用__next__()方法从iterator中检索下一个项目如果给定default，则如果迭代器为耗尽，否则StopIteration升高。

Answer 2

作为一个替代答案，您可以使用numpy.where而不是在数组上循环，并使用enumerate:

>>> import numpy as np
>>> def test(a,b):
...    try :
...         return np.where(a!=b)[0][0]
...    except IndexError:
...         return None
... 
>>> test(it1,it2)
>>> it1 = np.array([1,2,3,4])
>>> it2 = np.array([1,2,0,4])
>>> test(it1,it2)
2

或使用next:

>>> it1 = np.array([1,2,3,4,6])
>>> it2 = np.array([1,2,3,0,6])
>>> next(iter(np.where(it1!=it2)[0]),None)
3

此外，对于generators，您可以使用numpy.fromiter将生成器转换为numpy数组！

一些基准：

:~$ python -m timeit "from itertools import izip;it1=[1,2,3,4];it2=[1,2,10,20];next((pos for pos, (v1, v2) in enumerate(izip(it1, it2)) if v1 != v2), None)"
100000 loops, best of 3: 2.77 usec per loop
:~$ python -m timeit "import numpy as np;it1=[1,2,3,4];it2=[1,2,10,20];next(iter(np.where(it1!=it2)[0]),None)"
100000 loops, best of 3: 2.02 usec per loop

Answer 3

我一直在尝试内置过滤器，但python3过滤器可能会更好。

filter(lambda x: x if len(set(x)) else None, zip(it1, it2))
>>> [(3, 10), (4, 20)]

然而，我们没有得到差异的索引，只是以这种方式得到不同的项目。

所以我一直在想更多的问题，并检查相关的问题，但实际上，OP想要的是索引而不是结果。因此，修改

filter(lambda x:x[0] if len(set(x[1])) > 1 else None, enumerate(zip(it1,it2)))
>>> [(2, (3, 10)), (3, (4, 20))]

当然，这甚至不能接近上面的答案，所以我不建议在生产代码中这样做。只有好处？没有进口。以下是我的timeit结果：

python -m timeit "import numpy as np;it1=[1,2,3,4];it2=[1,2,10,20];next(iter(np.where(it1!=it2)[0]),None)"
100000 loops, best of 3: 2.02 usec per loop
10 loops, best of 3: 13.1 usec per loop
python -m timeit "from itertools import izip;it1=[1,2,3,4];it2=[1,2,10,20];next((pos for pos, (v1, v2) in enumerate(izip(it1, it2)) if v1 != v2), None)"
100000 loops, best of 3: 16.6 usec per loop
python -m timeit "it1=[1,2,3,4];it2=[1,2,10,20];filter(lambda x:x[0] if len(set(x[1])) > 1 else None, enumerate(zip(it1,it2)))"
10000 loops, best of 3: 19.1 usec per loop