如何使用 argparse 要求我的 python 脚本的参数是 0.0-1.0 之间的浮点数？

Question

我想在 Python 2.7 上使用argparse来要求我的脚本参数之一介于 0.0 和 1.0 之间。 argparse.add_argument()是否支持这个？

Answer 1

add_argument的type参数只需要是一个可调用对象，它接受一个字符串并返回一个转换后的值。 您可以围绕float编写一个包装器，以检查其值并在超出范围时引发错误。

def restricted_float(x):
    try:
        x = float(x)
    except ValueError:
        raise argparse.ArgumentTypeError("%r not a floating-point literal" % (x,))

    if x < 0.0 or x > 1.0:
        raise argparse.ArgumentTypeError("%r not in range [0.0, 1.0]"%(x,))
    return x

p = argparse.ArgumentParser()
p.add_argument("--arg", type=restricted_float)

Answer 2

这是一个使用choices参数add_argument ，其中自定义类被认为“等于”指定范围内的任何浮点数：

import argparse

class Range(object):
    def __init__(self, start, end):
        self.start = start
        self.end = end
    def __eq__(self, other):
        return self.start <= other <= self.end

parser = argparse.ArgumentParser()
parser.add_argument('--foo', type=float, choices=[Range(0.0, 1.0)])

Answer 3

添加str使边界在帮助中可见。

import argparse

class Range(object):
    def __init__(self, start, end):
        self.start = start
        self.end = end

    def __eq__(self, other):
        return self.start <= other <= self.end

    def __contains__(self, item):
        return self.__eq__(item)

    def __iter__(self):
        yield self

    def __str__(self):
        return '[{0},{1}]'.format(self.start, self.end)

parser = argparse.ArgumentParser()
parser.add_argument('--foo', type=float, choices=Range(0.0, 1.0))
parser.add_argument('--bar', type=float, choices=[Range(0.0, 1.0), Range(2.0,3.0)])

Answer 4

argparse.add_argument 调用需要一个可迭代的作为“选择”参数。 那么将 iterable 属性添加到上面的 Range 类呢？ 所以这两种情况都可以使用：

import argparse

class Range(object):
    def __init__(self, start, end):
        self.start = start
        self.end = end

    def __eq__(self, other):
        return self.start <= other <= self.end

    def __contains__(self, item):
        return self.__eq__(item)

    def __iter__(self):
        yield self

parser = argparse.ArgumentParser()
parser.add_argument('--foo', type=float, choices=Range(0.0, 1.0))
parser.add_argument('--bar', type=float, choices=[Range(0.0, 1.0), Range(2.0,3.0)])