将参数解析为字典 argparse

Question

我想将参数传递给 python 脚本，如下所示。

python test.py --env_a 5 --env_b 8

这样所有以“env”开头的参数都被添加到字典中，例如：

env = {
"a": 5,
"b": 8
}

当然，重点是这些参数的数量和名称不是预定义的。

您知道这是否可行或有更好的方法吗？

我正在使用 argparse 所以类似于

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--env-*", type=dict, default=0)
args = parser.parse_args()

print(args.env)

会很理想。

Answer 1

除了手动，我没有其他方法：

import argparse

parser = argparse.ArgumentParser()
args, unknown = parser.parse_known_args()

# unknown == ['--env_a', '5', '--env_b', '8']

env = {}
for i in range(0, len(unknown), 2):
    if not unknown[i].startswith('--env'): continue
    assert unknown[i+1].isdigit()
    env[unknown[i][len('--env_'):]] = int(unknown[i+1])

# env == {'a': 5, 'b': 8}

Answer 2

如果您可以指定如下参数：python argtest.py --env a 1 b 2 c hello，您可以使用use nargs='*'，然后使用您从中获得的列表创建字典

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--env", nargs='*', default='')
args = parser.parse_args()

print(args.env)
#Outputs 
# ['a', '1', 'b', '2']

keys = args.env[::2] # Every even element is a key
vals = args.env[1::2] # Every odd element is a value

# Make the dict you want
args.env = {k: int(v) if v.isdigit() else v for k, v in zip(keys, vals)} 
#####
# you can also write this as:
# 
# vals = [int(v) if v.isdigit() else v for v in args.env[1::2]]
# args.env = dict(zip(keys, vals))
#
####

print(args.env)
# Outputs
# {'a': '1', 'b': '2'}