Question

我想用多个参数解析一个选项并返回一个值作为我选择的集合，例如tuple或np.array而不是list。

我可以按照this answer，

中的建议，在解析后简单地转换结果列表

parser.add_argument('--foo', nargs='*')
options = parser.parse_args('--foo 1 2 3'.split())
# manual casts go here
options.foo = tuple(options.foo)

但这可能很麻烦且容易出错。如果我们有很多这样的参数要转换。

是否可以让ArgParser本身将参数存储在列表以外的集合中？

Answer 1

可以通过创建自定义操作来实现。例如，

import argparse
import numpy as np

def collect_as(coll_type):
  class Collect_as(argparse.Action):
    def __call__(self, parser, namespace, values, options_string=None):
      setattr(namespace, self.dest, coll_type(values))
  return Collect_as

parser = argparse.ArgumentParser()
parser.add_argument('--foo', nargs='*', type=int, action=collect_as(tuple))
parser.add_argument('--bar', nargs='*', type=np.float32, action=collect_as(np.array))
options = parser.parse_args('--foo 1 2 3 --bar 4 5 6'.split())
print(options)
# Namespace(bar=array([4., 5., 6.], dtype=float32), foo=(1, 2, 3))

使用nargs参数时，如何使argparse.ArgumentParser返回元组或np.array而不是列表？

1 个答案: