[英]Python argparse dict arg
我想从命令行接收一个dict(str -> str)
参数。 argparse.ArgumentParser
是否提供它? 或者其他图书馆?
对于命令行:
program.py --dict d --key key1 --value val1 --key key2 --value val2
我期待以下字典:
d = {"key1": "val1", "key2": "val2"}
这是使用自定义操作的另一个解决方案,如果您想将 dict 键对一起指定为逗号分隔 -
import argparse
import sys
parser = argparse.ArgumentParser(description='parse key pairs into a dictionary')
class StoreDictKeyPair(argparse.Action):
def __call__(self, parser, namespace, values, option_string=None):
my_dict = {}
for kv in values.split(","):
k,v = kv.split("=")
my_dict[k] = v
setattr(namespace, self.dest, my_dict)
parser.add_argument("--key_pairs", dest="my_dict", action=StoreDictKeyPair, metavar="KEY1=VAL1,KEY2=VAL2...")
args = parser.parse_args(sys.argv[1:])
print args
跑步:
python parse_kv.py --key_pairs 1=2,a=bbb,c=4 --key_pairs test=7,foo=bar
输出:
Namespace(my_dict={'1': '2', 'a': 'bbb', 'c': '4', 'test': '7', 'foo': 'bar'})
如果要在字符串中使用nargs而不是逗号分隔值:
class StoreDictKeyPair(argparse.Action):
def __init__(self, option_strings, dest, nargs=None, **kwargs):
self._nargs = nargs
super(StoreDictKeyPair, self).__init__(option_strings, dest, nargs=nargs, **kwargs)
def __call__(self, parser, namespace, values, option_string=None):
my_dict = {}
print "values: {}".format(values)
for kv in values:
k,v = kv.split("=")
my_dict[k] = v
setattr(namespace, self.dest, my_dict)
parser.add_argument("--key_pairs", dest="my_dict", action=StoreDictKeyPair, nargs="+", metavar="KEY=VAL")
args = parser.parse_args(sys.argv[1:])
print args
跑步
python arg_test4.py --key_pairs 1=2 a=bbb c=4 test=7 foo=bar
输出:
values: ['1=2', 'a=bbb', 'c=4', 'test=7', 'foo=bar']
Namespace(my_dict={'1': '2', 'a': 'bbb', 'c': '4', 'test': '7', 'foo': 'bar'})
我会使用这样的东西:
p = argparse.ArgumentParser()
p.add_argument("--keyvalue", action='append',
type=lambda kv: kv.split("="), dest='keyvalues')
args = p.parse_args("--keyvalue foo=6 --keyvalue bar=baz".split())
d = dict(args.keyvalues)
您可以创建一个自定义操作,将已解析的键值对直接“附加”到字典中,而不是简单地累积(key, value)
元组的列表。 (我看到的是 skyline75489 所做的;我的答案不同,使用带有自定义类型的单个--keyvalue
选项而不是单独的--key
和--value
选项来指定对。)
只是另一种简单的方法:
parser = argparse.ArgumentParser()
parser.add_argument('--key1')
parser.add_argument('--key2')
args = parser.parse_args()
my_dict = args.__dict__
Python 以数组argv
的形式接收参数。 您可以使用它在程序本身中创建字典。
import sys
my_dict = {}
for arg in sys.argv[1:]:
key, val=arg.split(':')[0], arg.split(':')[1]
my_dict[key]=val
print my_dict
对于命令行:
python program.py key1:val1 key2:val2 key3:val3
输出:
my_dict = {'key3': 'val3', 'key2': 'val2', 'key1': 'val1'}
注意:args 将在字符串中,因此您必须将它们转换为存储数值。
我希望它有帮助。
Python 单行 argparse 字典参数argparse_dictionary.py
# $ python argparse_dictionary.py --arg_dict=1=11,2=22;3=33 --arg_dict=a=,b,c=cc,=dd,=ee=,
# Namespace(arg_dict={'1': '11', '2': '22', '3': '33', 'a': '', 'c': 'cc', '': 'dd'})
import argparse
arg_parser = argparse.ArgumentParser()
arg_parser.add_argument(
'--arg_dict',
action=type(
'', (argparse.Action, ),
dict(__call__=lambda self, parser, namespace, values, option_string: getattr(
namespace, self.dest).update(
dict([
v.split('=') for v in values.replace(';', ',').split(',')
if len(v.split('=')) == 2
])))),
default={},
metavar='KEY1=VAL1,KEY2=VAL2;KEY3=VAL3...',
)
print(arg_parser.parse_args())
使用变量
d = vars(parser.parse_args())
解析输入的直接方法,如:
program.py --dict d --key key1 --value val1 --key key2 --value val2
是:
parser=argparse.ArgumentParser()
parser.add_argument('--dict')
parser.add_argument('--key', action='append')
parser.add_argument('--value', action='append')
args = parser.parse_args()
应该产生(如果我的心理解析器是正确的)
args = Namespace(dict='d', key=['key1','key2'], value=['value1','value2'])
您应该能够从中构建一个字典:
adict = {k:v for k, v in zip(args.key, args.value)}
使用args.dict
将其分配给具有该名称的变量需要一些非 Python 技巧。 最好的办法是在另一个字典中用这个名字创建一个元素。
another_dict = {args.dict: adict}
此解决方案不会执行太多错误检查。 例如,它不能确保键和值的数量相同。 它也不会让您创建多个字典(即重复的--dict
参数)。 它不需要任何特殊命令。 --dict
可能出现在--key key1
对之后。 几个--value
参数可以放在一起。
像chepner
那样将key=value
chepner
在一起可以解决许多这些问题。
如果您只是想将 argparse 输入转换为字典,那么 Python 3.6 中有一个简单的解决方案。 一个例子如下:
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('-i', '--input', help='the path to the input file')
parser.add_argument('-o', '--output', help='the path to the output file')
args = parser.parse_args()
arguments = dict(args._get_kwargs())
for k, v in arguments.items():
print(k, v)
给定命令行输入,例如python3 script_name.py --input 'input.txt' --output 'output.txt'
代码将输出到终端:
input input.txt
output output.txt
对于当前的库,如argparse 、 docopt和click ,它们都不支持使用dict
args。 我能想到的最好的解决方案是制作一个自定义的argparse.Action
来支持它自己:
import argparse
class MyAction(argparse.Action):
def __init__(self, option_strings, dest, nargs=None, **kwargs):
super(MyAction, self).__init__(option_strings, dest, nargs, **kwargs)
def __call__(self, parser, namespace, values, option_string=None):
print '%r %r %r' % (namespace, values, option_string)
value_dict = {}
values.reverse()
while len(values) > 0:
v = eval(values.pop()).lstrip('--') # This is like crazy hack, I know.
k = eval(values.pop())
value_dict[k] = v
setattr(namespace, self.dest, value_dict)
parser = argparse.ArgumentParser()
parser.add_argument('-d', action=MyAction, nargs='*')
args = parser.parse_args('-d "--key" "key1" "--value" "val1" "--key" "key2" "--value" "val2"'.split())
print(args)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.