如何实现进阶版Python any迭代器，返回匹配元素或索引

你需要的是一个能返回第一个匹配元素/索引而非仅布尔值的迭代工具，以下几种方案都能满足需求，且无需额外遍历循环，性能高效：

方案1：用next()直接获取首个匹配项 #

利用生成器表达式配合next()，找到匹配项后立即停止迭代，是最简洁高效的写法：

messages = [(0, None),(1, None), (2, "data")]

# 生成器筛选非None的元素，next()取第一个，无匹配则返回None
match = next((item for item in messages if item[1] is not None), None)

if match:
    id, data = match
    print(f"Tuple with data has id: {id} and data: '{data}'.")
else:
    print(f"There is no message with filled data")

输出：

Tuple with data has id: 2 and data: 'data'.

方案2：自定义复用型匹配函数 #

如果需要多次使用这类逻辑，可以封装成自定义函数，可读性和复用性更强：

def find_first_match(iterable, condition):
    """返回迭代器中第一个满足条件的元素，无匹配则返回None"""
    for item in iterable:
        if condition(item):
            return item
    return None

messages = [(0, None),(1, None), (2, "data")]

match = find_first_match(messages, lambda x: x[1] is not None)

if match:
    id, data = match
    print(f"Tuple with data has id: {id} and data: '{data}'.")
else:
    print(f"There is no message with filled data")

方案3：海象运算符实现 #

如果你偏好使用海象运算符:=，可以在生成器表达式中赋值，注意优先级需要用括号包裹：

messages = [(0, None),(1, None), (2, "data")]

match = None
# 每次迭代时将当前元素赋值给match，满足条件时any返回True
if any((match := item)[1] is not None for item in messages):
    id, data = match
    print(f"Tuple with data has id: {id} and data: '{data}'.")
else:
    print(f"There is no message with filled data")

以上三种方案均会在找到第一个匹配项后立即终止迭代，避免不必要的遍历，同时代码简洁易读，符合你的性能和可读性要求。

内容的提问来源于stack exchange，提问作者Call_me_Utka