Python命令行参数处理示例:接受用户输入并执行操作
发布时间:2023-12-04 09:42:58
Python中,我们可以使用sys模块来处理命令行参数。sys模块提供了argv变量,它是一个包含命令行参数的列表。argv[0]表示脚本名称,而argv[1:]表示除脚本名称之外的其他参数。
下面是一个示例,演示如何接受用户输入并执行相应操作:
import sys
# 检查命令行参数的数量
if len(sys.argv) < 2:
print("请输入参数")
sys.exit(1)
# 获取用户输入的命令行参数
command = sys.argv[1]
# 根据不同的参数执行不同的操作
if command == "hello":
print("Hello, World!")
elif command == "add":
if len(sys.argv) < 4:
print("请输入两个数字")
sys.exit(1)
num1 = float(sys.argv[2])
num2 = float(sys.argv[3])
result = num1 + num2
print(f"两个数字相加的结果是:{result}")
else:
print("无效的参数")
在这个例子中,我们首先检查命令行参数的数量,如果少于2个(包括脚本名称在内),则打印出一条错误信息并退出。然后,我们从命令行参数中获取用户输入的操作命令,并根据不同的命令执行相应的操作。
如果用户输入的命令是"hello",则打印出"Hello, World!";如果用户输入的命令是"add",则要求用户再输入两个数字,然后将这两个数字相加,并打印出结果。
让我们以不同的方式运行这个脚本来测试它:
$ python script.py hello Hello, World! $ python script.py add 5 7 两个数字相加的结果是:12 $ python script.py add 3.14 2.71 两个数字相加的结果是:5.85 $ python script.py 请输入参数 $ python script.py invalid 无效的参数
这个示例展示了如何使用sys模块来处理命令行参数,并根据用户的输入执行相应的操作。你可以根据自己的需求修改这个示例,并添加更多功能。
