欢迎访问宙启技术站
智能推送

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模块来处理命令行参数,并根据用户的输入执行相应的操作。你可以根据自己的需求修改这个示例,并添加更多功能。