在python 3中将字符串转换为数字列表

Question

这告诉我们准备数字列表的方法，包括一系列连续数字或非连续数字。

例如， list(range(1,10)) + [20,30,40] + list(range(400,410))将返回[1, 2, 3, 4, 5, 6, 7, 8, 9, 20, 30, 40, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409] 。

但是，仅输入"1-10, 20, 30, 40, 400-410"这样的字符串，我可以使其更简单吗？

我可以使用for循环编写一个函数来遍历数字。 是否有更有效的方式编写函数而不使用for循环？

Answer 1

您可以使用嵌套列表推导，检查当前零件是否包含-然后使用range或相应地创建一个单元素列表：

>>> s = "1-10, 20, 30, 40, 400-410"
>>> [n for part in s.split(", ") for n in (range(*map(int, part.split("-"))) if "-" in part else [int(part)])]
[1, 2, 3, 4, 5, 6, 7, 8, 9, 20, 30, 40, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409]

出于可读性考虑，也许可以将其拆分：

>>> to_list = lambda part: range(*map(int, part.split("-"))) if "-" in part else [int(part)]
>>> [n for part in s.split(", ") for n in to_list(part)]

注意：与第一个示例一样，这会将"1-10"转换为[1, 2, ..., 9] ，而不包含10 。

---

如评论中所述，这不适用于负数，但是尝试将-3或-4--2为成对的数字。 为此，您可以使用正则表达式...

>>> def to_list(part):
...     m =re.findall(r"(-?\d+)-(-?\d+)", part)
...     return range(*map(int, m[0])) if m else [int(part)]
...
>>> s = "-10--5, -4, -2-3"
>>> [n for part in s.split(", ") for n in to_list(part)]
[-10, -9, -8, -7, -6, -4, -2, -1, 0, 1, 2]

...或仅对范围使用不同的分隔符，例如-10:-5 。

Answer 2

我建议您改写一个允许创建更复杂的分段范围的函数。

def piecewise_range(*args):
    for arg in args:
        yield from range(*arg)

piecewise_range((1, 10), (20, 50, 10), (400, 410))

然后，函数piecewise_range输出一个生成器，因此它使您可以使用更复杂的范围模式并保留range延迟。

Answer 3

根据您输入字符串的来源，您可以使用如下所示的名称进行命名：

import re

def gen_nums(input_string):
    out = []
    for item in re.split(r',\s*', input_string):
        if '-' in item:
            start, end = item.split('-')
            out.extend(range(int(start), int(end)))
        else:
            out.append(int(item))
    return out