将参数解析为字典as解析

Question

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

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

因此，所有以"env“开头的参数都添加到字典中，如下所示：

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

当然，问题的关键是，这些参数的数目和名称并不是预先定义的。

你知道这是可能的还是有更好的方法来做到这一点？

我使用的是am解析，所以类似于

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

print(args.env)

会是理想的。

Answer 1

如果您可以指定这样的参数：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'}

Answer 2

除了手工操作之外，我看不出任何其他方法：

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}