📌  相关文章
📜  如何在 python 中获取未知数量的输入(1)

📅  最后修改于: 2023-12-03 15:24:20.164000             🧑  作者: Mango

如何在 python 中获取未知数量的输入

在实际编程中,我们有时需要从用户获取未知数量的输入,例如用户可以输入一个整数,也可以输入多个整数以空格分隔。本文将介绍获取未知数量的输入的几种方法。

方法1:使用 input() 函数循环获取输入

使用 input() 函数获取单个输入是很容易的,我们只需将其赋值给一个变量即可。但是如果需要获取未知数量的输入,我们可以使用 while 循环不断获取输入,直到用户输入一个指定字符来结束输入。

inputs = []
while True:
    input_str = input("请输入(输入'q'结束输入):")
    if input_str == 'q':
        break
    inputs.append(input_str)
print(inputs)

以上代码中,我们使用了一个空列表 inputs 来存储每次获取的输入。通过 while 循环不断获取输入,当用户输入字符 'q' 时退出循环,将获取的所有输入存储在列表中。

方法2:使用 split() 函数分割输入

如果用户输入的每个元素之间用空格分隔,我们可以使用 split() 函数将输入字符串分割成多个元素。使用 split() 函数得到的是一个列表,我们可以使用 *args 来接收其中的每个元素。

inputs = input("请输入:").split()
print(inputs)

以上代码中,我们使用了一个空格作为分隔符来分割输入字符串,得到一个包含所有输入的列表。使用 print() 函数打印列表即可。

方法3:使用 argparse 库获取命令行输入

如果我们需要从命令行获取输入,可以使用 argparse 库来简化输入的管理。argparse 库提供了诸如列表、布尔、整数等多种参数类型,使得我们可以轻松处理各种命令行输入。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("--inputs", nargs="+", type=int, help="输入整数列表")
args = parser.parse_args()

inputs = args.inputs
print(inputs)

以上代码中,我们使用 argparse 库来管理命令行参数。通过 add_argument() 函数添加 --inputs 参数,其 nargs 参数设置为 '+' 表示我们可以接收多个输入。类型设置为整数后,argparse 库将自动将输入转换为整数并存储在 inputs 参数中。使用 print() 函数打印即可。

总结:以上就是获取未知数量输入的几种方法,不同的方法有着各自的适用场景。在实际编程中,需要根据具体需求选择最适合的方法。