python - python3中的惰性切片

Question

我在 python 中有一个巨大的列表(~1_800_000 项)，它是使用 map() 函数从大约 1000 个 JSON 文件构建的。我想检查几个第一项以确保脚本正常工作。我是这样做的:

items = map(lambda file: load_json(file), file_list)
print(list(items)[:5])

将 map 转换为 list 大约需要 5-10 秒，是否可以在不将 map 结果转换为 list 的情况下取几个第一个项目？

Answer 1

你可以这样做:

items = map(lambda file: load_json(file), file_list)
print([next(items) for _ in range(5)])

或者使用itertools.islice ，它有一点好处，如果你的项目少于五个，它不会失败:

items = map(lambda file: load_json(file), file_list)
print(list(itertools.islice(items, 5)))

请注意，这两个都使用 items 中的第一个元素，因此如果您想“查看”这些元素然后获取整个列表，您需要先添加这些项目。