Python处理超长输入: 无需外部库的4种方法

2024-03-14 02:22:27

如何在 Python 中处理超长输入

引言

作为一位经验丰富的程序员，我时常会遇到需要处理超长输入的挑战。Python 的 input() 函数通常用于从用户那里获取输入，但它有一个限制，即截断超过一定长度的输入。这可能会给我们带来问题，尤其是当我们处理超大的文本文件或数据集时。

为了克服这一限制，我探索了多种方法，而不依赖额外的 Python 库。在这篇文章中，我将分享我发现的解决方法，并为每种方法提供详细的解释和代码示例。

解决方法

1. 使用 sys.stdin.read()

sys.stdin.read() 函数从标准输入流中读取所有内容，不受长度限制。我们可以使用它来检索输入的完整内容。

import sys

input_text = sys.stdin.read()
print(len(input_text))

2. 分块读取

我们可以将超长输入分解为较小的块，然后逐块读取。以下代码将输入划分为 16,000 个字符的块，然后将其全部连接起来。

input_text = ""

while True:
    chunk = input(16000)
    if not chunk:
        break
    input_text += chunk

print(len(input_text))

3. 使用生成器函数

生成器函数可以逐个字符地生成输入，而无需将整个输入存储在内存中。我们可以使用它来迭代输入。

def input_generator():
    while True:
        chunk = input(16000)
        if not chunk:
            return
        yield chunk

for chunk in input_generator():
    # 处理 chunk
    pass

4. 读取为单词

尽管 Python 的 input() 函数不会将输入读取为单个单词，但我们可以使用 itertools.chain() 函数将输入拆分为单词。

from itertools import chain

input_words = chain(*map(lambda line: line.split(), input().splitlines()))

for word in input_words:
    # 处理 word
    pass