如何从嵌套列表中获取前n个首元素等于'value'的元素？

这里提供几种更简洁高效的实现方式，替代原有的两步操作：

1. 生成器+itertools.islice（内存友好，推荐） #

这种方式无需提前存储所有符合条件的元素，而是边遍历边筛选，取到前n个后立即停止，非常适合处理大规模数据：

from itertools import islice

items = [['value', 0], ['foo', 1], ['value', 2], ['value', 3], ['bar', 4]]
n = 2

# 生成器表达式筛选符合条件的元素，islice截取前n个
filtered_items = (item for item in items if item[0] == 'value')
for item in islice(filtered_items, n):
    print(item)

2. 列表推导式+切片（代码极简） #

如果数据量不大，直接用列表推导式生成符合条件的列表，再通过切片获取前n个，代码更简洁直观：

items = [['value', 0], ['foo', 1], ['value', 2], ['value', 3], ['bar', 4]]
n = 2

# 一步生成前n个符合条件的元素列表
target_items = [item for item in items if item[0] == 'value'][:n]
for item in target_items:
    print(item)

3. 单次遍历计数（无依赖实现） #

不想引入额外模块的话，可以在一次遍历中完成筛选和计数，找到目标数量后直接退出循环：

items = [['value', 0], ['foo', 1], ['value', 2], ['value', 3], ['bar', 4]]
n = 2

count = 0
for item in items:
    if item[0] == 'value':
        print(item)
        count += 1
        if count == n:
            break

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