gpt4 book ai didi

python - numpy 数组的 ArgumentParser

转载 作者:行者123 更新时间:2023-12-02 03:32:01 25 4
gpt4 key购买 nike

有没有办法为 np.array 而不是列表添加参数到 ArgumentParser ?我知道我可以做这样的事情

import argparse
parser = argparse.ArgumentParser(prog='PROG')
parser.add_argument('-foo', action='store', type=int, nargs='+')
args = parser.parse_args(['-foo', '1', '2'])
args.foo = np.array(args.foo)

但我想在解析之前指定参数的完整描述。

有办法做到这一点吗?

最佳答案

要指定 StoreAction 操作处理程序工作方式的细微变化,您需要创建一个 subclass进行适当更改的处理程序(文档在上面 this section 有一个示例)

import argparse, numpy as np

class Store_as_array(argparse._StoreAction):
def __call__(self, parser, namespace, values, option_string=None):
values = np.array(values)
return super().__call__(parser, namespace, values, option_string)

parser = argparse.ArgumentParser(prog='PROG')
parser.add_argument('-foo', action=Store_as_array, type=int, nargs='+')
# ^ specify as the action
args = parser.parse_args(['-foo', '1', '2'])

assert isinstance(args.foo, np.ndarray)

关于python - numpy 数组的 ArgumentParser,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37755273/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com