使用Python的argparse如何接受形如0-5的数字范围？

Question

使用argparse，有没有一种方法可以接受数字范围并将其转换为列表？

例如：

python example.py --range 0-5

有没有一种方法可以以这种形式输入命令行参数，并最终得到：

args.range = [0,1,2,3,4,5]

还有可能输入--range 2 = [2]吗？

- Takkun

---

我怀疑这样的东西是否内置。不过让 argparse 接受任何字符串并自行解析相当简单。 - user395760

Answer 1

你可以在type参数中编写自己的解析器，例如：

from argparse import ArgumentParser, ArgumentTypeError
import re

def parseNumList(string):
    m = re.match(r'(\d+)(?:-(\d+))?$', string)
    # ^ (or use .split('-'). anyway you like.)
    if not m:
        raise ArgumentTypeError("'" + string + "' is not a range of number. Expected forms like '0-5' or '2'.")
    start = m.group(1)
    end = m.group(2) or start
    return list(range(int(start,10), int(end,10)+1))

parser = ArgumentParser()
parser.add_argument('--range', type=parseNumList)

args = parser.parse_args()
print(args)

~$ python3 z.py --range m
usage: z.py [-h] [--range RANGE]
z.py: error: argument --range: 'm' is not a range of number. Expected forms like '0-5' or '2'.

~$ python3 z.py --range 2m
usage: z.py [-h] [--range RANGE]
z.py: error: argument --range: '2m' is not a range of number. Expected forms like '0-5' or '2'.

~$ python3 z.py --range 25
Namespace(range=[25])

~$ python3 z.py --range 2-5
Namespace(range=[2, 3, 4, 5])

- kennytm

Answer 2

你可以只使用一个字符串参数，然后使用range(*rangeStr.split(','))进行解析。

- phihag