如何实现Python脚本支持可选命令行参数修改初始值？

嘿，这个问题我太熟了——直接按索引取sys.argv确实会踩坑，毕竟用户输入参数的顺序可能乱，还可能跳过某些参数。给你几个实用的解决思路，从简单到规范都有：

这种方法不用依赖任何额外库，自己处理键值对就行，特别适合参数不多的小脚本。核心思路是：先定义好默认值，然后遍历所有传入的参数，把每个KEY=VALUE的字符串拆成键和值，再更新对应的变量。

举个完整的例子：

# 先定义好所有默认参数
SIZE = 100
NB = 100
WIDTH_CANVAS = 150

import sys

# 跳过第一个参数（脚本本身的名字），遍历剩下的所有参数
for arg in sys.argv[1:]:
    # 用等号分割键和值，这里加个异常处理防止用户输入格式错误
    try:
        key, value = arg.split('=')
    except ValueError:
        print(f"忽略无效参数：{arg}，请使用KEY=VALUE的格式")
        continue
    
    # 根据键更新对应变量，记得把字符串转成整数（因为你的参数都是数字）
    if key == 'SIZE':
        SIZE = int(value)
    elif key == 'NB':
        NB = int(value)
    elif key == 'WIDTH_CANVAS':
        WIDTH_CANVAS = int(value)
    else:
        print(f"警告：未知参数 {key}，已忽略")

# 测试一下效果
print(f"最终参数：SIZE={SIZE}, NB={NB}, WIDTH_CANVAS={WIDTH_CANVAS}")

这样用户不管输入什么顺序的参数，比如python3 main.py NB=10 SIZE=50，或者只输入python3 main.py WIDTH_CANVAS=200，甚至不输入任何参数，都能正常工作。

如果你的脚本以后可能扩展更多参数，用argparse更规范——它是Python自带的标准库，能自动处理参数类型、生成帮助信息，还能帮你校验参数格式。

示例代码：

import argparse

# 创建一个参数解析器，还可以加个描述说明
parser = argparse.ArgumentParser(description='配置程序的初始参数')

# 逐个添加参数，指定默认值、类型和帮助文本
parser.add_argument('--SIZE', type=int, default=100, help='画布元素大小，默认值100')
parser.add_argument('--NB', type=int, default=100, help='元素数量，默认值100')
parser.add_argument('--WIDTH_CANVAS', type=int, default=150, help='画布宽度，默认值150')

# 解析命令行参数
args = parser.parse_args()

# 把解析后的参数赋值给变量（或者直接用args.SIZE这种形式）
SIZE = args.SIZE
NB = args.NB
WIDTH_CANVAS = args.WIDTH_CANVAS

print(f"最终参数：SIZE={SIZE}, NB={NB}, WIDTH_CANVAS={WIDTH_CANVAS}")

用法稍微有点不同，用户需要用--前缀来指定参数，比如：

python3 main.py --NB=10 --WIDTH_CANVAS=200

如果用户想查看帮助，直接输入python3 main.py -h就能看到自动生成的帮助文档，非常友好。

如果你的项目需要更强大的命令行交互（比如短选项、颜色输出、子命令等），可以试试click库——它比argparse更灵活，写法也更简洁。

首先需要安装：

pip install click

然后写代码：

import click

# 用装饰器定义命令和参数
@click.command()
@click.option('--SIZE', default=100, type=int, help='画布元素大小，默认值100')
@click.option('--NB', default=100, type=int, help='元素数量，默认值100')
@click.option('--WIDTH_CANVAS', default=150, type=int, help='画布宽度，默认值150')
def main(SIZE, NB, WIDTH_CANVAS):
    # 这里写你的主程序逻辑
    print(f"最终参数：SIZE={SIZE}, NB={NB}, WIDTH_CANVAS={WIDTH_CANVAS}")

if __name__ == '__main__':
    main()

用法和argparse类似，但click的帮助信息更美观，还支持短选项（比如可以把--SIZE改成-s），适合做更专业的命令行工具。

总结一下：

• 小脚本、参数少：用手动解析
• 正规项目、需要规范：用argparse（不用额外安装）
• 复杂命令行交互：用click

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