Python中使用sys.argv获取参数时如何为函数设置可选参数

错误原因 #

你的报错根源是Python执行函数调用前会先计算所有入参的取值，再传入函数。当你没有传入第三个命令行参数时，sys.argv列表只有3个元素（下标0对应脚本名，1对应第一个参数，2对应第二个参数），代码在执行sys.argv[3]取值的步骤就直接抛出下标越界错误，根本不会进入main函数的默认参数判断逻辑。

解决方法 #

你可以先判断sys.argv的长度，再决定是否传入第三个参数，以下是几种常用实现方式： #

写法1：条件判断传参

import sys

def main(w, l, s=None):
    variable_1 = w
    variable_2 = l
    variable_3 = s
    # 后续业务逻辑

if __name__ == "__main__":
    # 先校验必选参数是否足够，提前给出友好提示
    if len(sys.argv) < 3:
        print("用法：python 脚本名.py 宽度 长度 [可选字符串参数]")
        sys.exit(1)
    w = int(sys.argv[1])
    l = int(sys.argv[2])
    if len(sys.argv) >= 4:
        main(w, l, sys.argv[3])
    else:
        main(w, l)

写法2：三元表达式简化写法

import sys

def main(w, l, s=None):
    variable_1 = w
    variable_2 = l
    variable_3 = s
    # 后续业务逻辑

if __name__ == "__main__":
    if len(sys.argv) < 3:
        print("用法：python 脚本名.py 宽度 长度 [可选字符串参数]")
        sys.exit(1)
    main(
        int(sys.argv[1]),
        int(sys.argv[2]),
        sys.argv[3] if len(sys.argv) >= 4 else None
    )

正式工具推荐写法：用标准库argparse处理参数

如果是需要对外使用的命令行工具，更推荐用Python内置的argparse模块，它会自动处理参数校验、帮助提示等逻辑，不需要手动判断列表长度：

import argparse

def main(w, l, s=None):
    variable_1 = w
    variable_2 = l
    variable_3 = s
    # 后续业务逻辑

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("w", type=int, help="宽度参数")
    parser.add_argument("l", type=int, help="长度参数")
    parser.add_argument("s", nargs="?", default=None, help="可选字符串参数")
    args = parser.parse_args()
    main(args.w, args.l, args.s)

内容的提问来源于stack exchange，提问作者user15064722