温馨提示×

python命令行参数能自定义吗

小樊
81
2024-11-18 15:14:09
栏目: 编程语言

是的,Python命令行参数可以自定义

import sys

def main(arg1, arg2):
    print(f"参数1: {arg1}")
    print(f"参数2: {arg2}")

if __name__ == "__main__":
    # 获取命令行参数,第一个参数是脚本名称,所以从索引1开始
    arg1 = sys.argv[1]
    arg2 = sys.argv[2]

    main(arg1, arg2)

在这个例子中,我们定义了一个名为main的函数,它接受两个参数。在if __name__ == "__main__"代码块中,我们从sys.argv列表中获取命令行参数。sys.argv[0]是脚本名称,所以我们从索引1开始获取参数。

现在,当你运行这个脚本并提供命令行参数时,例如:

python script.py 参数1 参数2

它将输出:

参数1: 参数1
参数2: 参数2

你可以根据需要自定义这些参数。

0