Question

我正在编写命令行应用程序，并希望用户能够将数字作为单个数字或作为范围输入。所以，例如：

$ myapp -n 3,4,5,6

或

$ myapp -n 3-6

我希望我的应用将这些放入Python列表中，例如，[3,4,5,6] 我正在使用optparse，但我不确定如何从这两种输入样式创建列表。一些示例代码会很棒。

修改

我希望能够输入多个范围：

$ myapp -n 22-27, 51-64

Answer 1

import argparse

def parse_range(astr):
    result = set()
    for part in astr.split(','):
        x = part.split('-')
        result.update(range(int(x[0]), int(x[-1]) + 1))
    return sorted(result)

parser = argparse.ArgumentParser()
parser.add_argument('-n', type=parse_range)
args = parser.parse_args()
print(args.n)

产量

% script.py -n 3-6
[3, 4, 5, 6]

% script.py -n 3,6
[3, 6]

% script.py -n 22-27,51-64 
[22, 23, 24, 25, 26, 27, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64]

Answer 2

如果你在字符串中有n-arg，你可以这样做：

def f(s):
    if '-' in s:
        i = s.index('-')
        return range(*map(int, s.split('-')))
    return map(int, s.split(','))

一些例子：

In [3]: s = '1, 2, 3, 4, 5, 6'

In [4]: f(s)
Out[4]: [1, 2, 3, 4, 5, 6]

In [5]: f('3-6')
Out[5]: [3, 4, 5]

In [6]: f('3-16-3')
Out[6]: [3, 6, 9, 12, 15]

Answer 3

您可以定义参数并使用optparse callback处理输入，然后再保存：

from optparse import OptionParser

parser = OptionParser()

def create_range_callback(option, opt, value, parser):
    i, j = map(int, value.split('-'))
    setattr(parser.values, option.dest, range(i, j+1))

parser.add_option("-r", "--range", action="callback", 
                  callback=create_range_callback, 
                  type="string", dest='list')

(options, args) = parser.parse_args()

print options.list

立即行动：

python2.7 test.py -r 1-5

输出：

[1, 2, 3, 4, 5]

Answer 4

使用Python's range function。解析用户的输入，用“ - ”字符拆分，然后将这些参数传递给范围。

您的代码可能如下所示：

parameters = input.split('-')
completeRange = range(int(parameters[0]), int(parameters[1]))

如果他们输入个别数字，您可以很容易地将其解析为一个列表。

Answer 5

您可以使用optparse库。

示例：

from optparse import OptionParser

opt_parser = OptionParser(version="%prog 0.1")
opt_parser.usage = '%prog [options]\n\nTCP protocol reengineering tool'

# Options
opt_parser.add_option('-n', default="1,2,3")
(options, args) = opt_parser.parse_args()

list = []
for s in options.n.split(","):
    list.append(int(s))

解析数字的命令行输入

5 个答案: