将参数解析为字典 argparse

如何解决将参数解析为字典 argparse

我想将参数传递给 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)

将是理想的。

解决方法

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

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}

,

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