温馨提示×

python读取命令行参数案例

小樊
82
2024-11-28 01:46:25
栏目: 编程语言

以下是一个简单的 Python 程序,用于读取命令行参数并根据参数执行不同的操作:

import sys

def main():
    # 获取命令行参数的数量
    num_args = len(sys.argv)

    # 检查是否有足够的参数
    if num_args < 2:
        print("请提供至少一个命令行参数。")
        sys.exit(1)

    # 获取第一个参数(脚本名称除外)
    arg1 = sys.argv[1]

    # 根据第一个参数执行不同的操作
    if arg1 == "hello":
        print("你好!")
    elif arg1 == "goodbye":
        print("再见!")
    else:
        print(f"未知命令:{arg1}")

if __name__ == "__main__":
    main()

将此代码保存为 example.py,然后在命令行中运行它,如下所示:

python example.py hello

输出将是:

你好!

如果运行以下命令:

python example.py goodbye

输出将是:

再见!

如果运行以下命令:

python example.py unknown

输出将是:

未知命令:unknown

0