如何使用argparse动态收集前缀为env的命令行参数并转换为字典？

你好呀～你想要的这种动态收集以env-开头的命令行参数的需求，用标准的argparse直接写--env-*是行不通的，因为argparse的add_argument不支持这种通配符式的参数定义。不过有两种挺实用的实现方案，我给你详细说说：

这种方法更贴合argparse的使用习惯，通过自定义动作来捕获所有符合规则的参数：

import argparse

class EnvArgAction(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        # 从参数名里提取后缀（比如从"--env-a"提取"a"）
        key = option_string.lstrip('--').split('-', 1)[1]
        # 初始化namespace里的env字典，如果还没创建的话
        if not hasattr(namespace, 'env'):
            setattr(namespace, 'env', {})
        # 将键值对加入字典
        namespace.env[key] = values

parser = argparse.ArgumentParser()
# 添加一个可变参数占位符，所有--env-*格式的参数都会被这个Action处理
parser.add_argument('--env-*', action=EnvArgAction, type=int)
args = parser.parse_args()

print(args.env)

当你运行python test.py --env-a 5 --env-b 8时，脚本会输出{'a': 5, 'b': 8}。这里的核心是自定义的EnvArgAction会自动处理每个匹配--env-前缀的参数，提取后面的键名并组装成字典，完全满足你“参数数量和名称不预先定义”的需求。

如果觉得自定义Action有点繁琐，也可以直接手动遍历命令行参数，过滤出符合规则的项来组装字典：

import sys

env = {}
# 从第1个元素开始遍历（第0个是脚本文件名）
i = 1
while i < len(sys.argv):
    arg = sys.argv[i]
    if arg.startswith('--env-'):
        # 提取键名
        key = arg.lstrip('--').split('-', 1)[1]
        # 获取对应的值，注意要跳过下一个元素
        value = int(sys.argv[i+1])
        env[key] = value
        i += 2
    else:
        # 如果有其他参数，可在这里单独处理
        i += 1

print(env)

运行同样的命令，也会得到{'a': 5, 'b': 8}。这种方法更灵活，不需要依赖argparse的机制，适合参数规则比较简单的场景。

如果你的脚本还有其他常规参数需要用argparse处理，也可以结合两种方法：先用argparse处理固定参数，再手动遍历剩下的参数来收集env相关项，或者在自定义Action里同时处理其他逻辑。

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