单行检查迭代器是否至少产生一个元素？

Question

目前我正在这样做：

try:
    something = iterator.next()
    # ...
except StopIteration:
    # ...

但是我想要一个可以放在简单if语句中的表达式。 有什么内置的东西可以让这段代码看起来不那么笨拙吗？

如果可迭代对象为空，则any()返回False ，但如果不是，它可能会遍历所有项目。 我只需要它来检查第一项。

---

有人问我想做什么。 我编写了一个执行 SQL 查询并产生其结果的函数。 有时，当我调用此函数时，我只想知道查询是否返回任何内容并基于此做出决定。

Answer 1

如果为 True， any都不会超出第一个元素。 如果迭代器产生一些错误的东西，你可以写any(True for _ in iterator) 。

Answer 2

在 Python 2.6+ 中，如果 name sentinel绑定到迭代器不可能产生的值，

if next(iterator, sentinel) is sentinel:
    print('iterator was empty')

如果您不知道迭代器可能会产生什么，请使用自己的标记（例如在模块顶部）

sentinel = object()

否则，您可以在哨兵角色中使用迭代器不可能产生的任何您“知道”（基于应用程序考虑）的值。

Answer 3

这并不是很干净，但它显示了一种将其无损打包在函数中的方法：

def has_elements(iter):
  from itertools import tee
  iter, any_check = tee(iter)
  try:
    any_check.next()
    return True, iter
  except StopIteration:
    return False, iter

has_el, iter = has_elements(iter)
if has_el:
  # not empty

这不是真正的pythonic，对于特定情况，可能有更好（但不太通用）的解决方案，例如下一个默认值。

first = next(iter, None)
if first:
  # Do something

这并不普遍，因为 None 可以是许多迭代中的有效元素。

Answer 4

要做到这一点，最好的办法是用peekable从more_itertools 。

from more_itertools import peekable
iterator = peekable(iterator)
if iterator:
    # Iterator is non-empty.
else:
    # Iterator is empty.

请注意，如果您保留对旧迭代器的引用，则该迭代器将变得先进。 从那时起，您必须使用新的 peekable 迭代器。 但是，确实，peekable 期望是修改旧迭代器的唯一代码，因此无论如何您都不应该保留对旧迭代器的引用。

Answer 5

您可以使用：

if zip([None], iterator):
    # ...
else:
    # ...

但这对代码阅读器来说有点不解释

Answer 6

关于什么：

In [1]: i=iter([])

In [2]: bool(next(i,False))
Out[2]: False

In [3]: i=iter([1])

In [4]: bool(next(i,False))
Out[4]: True

Answer 7

这是一个矫枉过正的迭代器包装器，通常允许检查是否有下一个项目（通过转换为布尔值）。 当然效率很低。

class LookaheadIterator ():

    def __init__(self, iterator):
        self.__iterator = iterator
        try:
            self.__next      = next (iterator)
            self.__have_next = True
        except StopIteration:
            self.__have_next = False

    def __iter__(self):
        return self

    def next (self):
        if self.__have_next:
            result = self.__next
            try:
                self.__next      = next (self.__iterator)
                self.__have_next = True
            except StopIteration:
                self.__have_next = False

            return result

        else:
            raise StopIteration

    def __nonzero__(self):
        return self.__have_next

x = LookaheadIterator (iter ([]))
print bool (x)
print list (x)

x = LookaheadIterator (iter ([1, 2, 3]))
print bool (x)
print list (x)

输出：

False
[]
True
[1, 2, 3]

Answer 8

__length_hint__估计list(it)的长度 - 但它是私有方法：

x = iter( (1, 2, 3) )
help(x.__length_hint__)
      1 Help on built-in function __length_hint__:
      2 
      3 __length_hint__(...)
      4     Private method returning an estimate of len(list(it)).

Answer 9

有点晚了，但是...您可以将迭代器转换为列表，然后使用该列表：

# Create a list of objects but runs out the iterator.
l = [_ for _ in iterator]

# If the list is not empty then the iterator had elements; else it was empty.
if l :
    pass # Use the elements of the list (i.e. from the iterator)
else :
    pass # Iterator was empty, thus list is empty.