確定生成器的空性

在Python中，生成器是用於惰性評估和迭代大型資料集的強大工具。然而，由於發電機的動態特性，確定發電機是否為空是一個獨特的挑戰。與列表或元組不同，生成器沒有大小屬性，簡單地迭代它們會導致 StopIteration 異常。

Peek 與 HasNext

有些人可能會建議使用 peek() 函數檢查生成器是否有項目。雖然 peek() 允許您檢索生成器的第一項而不消耗它，但它沒有提供確定生成器是否為空的方法。這是因為 peek() 在生成器為空時引發 StopIteration 異常，這與嘗試迭代空生成器時引發的異常沒有區別。

類似地，hasNext() 函數通常在 Java 中找到，在 Python 產生器中不可用。

從Start

要確定生成器是否從一開始就是空的，您可以使用自訂函數，如響應中提供的函數：

def peek(iterable):
    try:
        first = next(iterable)
    except StopIteration:
        return None
    return first, itertools.chain([first], iterable)

此函數檢索產生器的第一個元素，如果生成器為空則傳回 None 。它還透過傳回包含第一個元素的修改版本來保留生成器的狀態。

用法

要使用此函數，您可以將生成器作為參數：

res = peek(mysequence)
if res is None:
    # sequence is empty.  Do stuff.
else:
    first, mysequence = res
    # Do something with first, maybe?
    # Then iterate over the sequence:
    for element in mysequence:
        # etc.

此檢查可讓您在迭代之前確定生成器是否為空並適當處理空情況。