您可以使用的
type=和
choices=参数
add_argument。仅接受
'0'和
'1',您可以这样做:
parser.add_argument(…, choices={"0", "1"})并且只接受整数,您可以这样做:
parser.add_argument(…, type=int)
请注意,在中
choices,您必须以指定为
type参数的类型提供选项。因此,要检查整数 并 只允许
0和
1,您可以执行以下操作:
parser.add_argument(…, type=int, choices={0, 1})例:
>>> import argparse>>> parser = argparse.ArgumentParser()>>> _ = parser.add_argument("-p", type=int, choices={0, 1})>>> parser.parse_args(["-p", "0"])Namespace(p=0)


