Python 使用 ArgumentParser 将 List[str] 转换为 Dict[str, str]

Question

我需要解析命令行参数 - 将 List[str] 转换为 Dict[str, str]

['key1=value1', 'key2=value2'] -> {key1: value1, key2:value2}

我使用以下代码：

 parser.add_argument(
        '--model-config',
        nargs='+',
        type=model_config_str_to_dict        
    )

其中 model_config_str_to_dict(model_config: List[ str]) 是将列表转换为字典的函数。但我得到的是 model_config 参数的 str 而不是 List[str] 。

我的问题是如何解析 --model-config 参数以将其转换为字典？

Answer 1

type 参数应用于 --model-config 之后的每个输入之前它们被附加到存储在解析的列表中参数作为属性 .model_config。换句话说，model_config 的值将是 List[model_config_str_to_dict(mc_arg)]，而不是 model_config_str_to_dict(List[mc_arg])

作为示例：

import argparse

def model_config_split(s):
    k, v = s.split('=', 1)
    return (k, v)

parser = argparse.ArgumentParser()
parser.add_argument( '--model-config',
        nargs='+',
        type=model_config_str_to_dict)

args = parser.parse_args(['--model-config', 'key1=value1', 'key2=value2'])
print(args.model_config)

# prints:
# [('key1', 'value1'), ('key2', 'value2')]

如果您想将其转换为字典，您可以在解析参数之后执行此操作。

args.model_config = dict(args.model_config)
print(args.model_config)

# prints:
# {'key1': 'value1', 'key2': 'value2'}