python 重定向标准输入

原创

mob64ca12f6aae1 2024-06-11 05:46:53 ©著作权

文章标签 标准输入重定向 Python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12f6aae1的原创作品，请联系作者获取转载授权，否则将追究法律责任

重定向标准输入的Python技巧

在Python中，我们通常使用input()函数来获取用户的输入，这时候标准输入（stdin）会默认从键盘获取输入。但有时候我们希望程序可以从文件或者其他来源获取输入，这时就需要重定向标准输入。

什么是重定向标准输入？

重定向标准输入指的是改变程序默认从键盘读取输入的行为，转而从另一个文件或者其他来源获取输入。这样可以方便测试和调试程序，也可以实现一些特定的功能。

如何重定向标准输入？

在Python中，可以使用sys.stdin来实现重定向标准输入。首先需要导入sys模块，然后通过sys.stdin改变标准输入的来源。

下面是一个示例代码，演示如何将标准输入从键盘重定向到一个文件中：

import sys

# 打开一个文件，准备作为标准输入
with open('input.txt', 'r') as f:
    # 保存原始的标准输入
    stdin_backup = sys.stdin

    # 将文件对象赋值给标准输入
    sys.stdin = f

    # 使用input()函数读取标准输入
    user_input = input("请输入内容：")
    print("你输入的内容是：", user_input)

    # 恢复原始的标准输入
    sys.stdin = stdin_backup

在这个示例中，我们将标准输入从键盘重定向到一个名为input.txt的文件中，并使用input()函数读取了文件中的内容。