我使用生成器表达式有什么问题？

Question

我有以下代码，其中我尝试将范围的字符串表示转换为数字列表。例如，如果输入为“0-0,3-5,7-10”，则预期输出为“0,3,4,5,7,8,9,10”。但是，我在以下位置出错：

for l,h in r.split('-') 

它说没有足够的值来解压。我的推理是，r应该具有'x-y'的形式，所以如果我将其拆分为'-'，我应该得到两个值。有什么问题吗？

def parse_ranges(ranges):
    """Return a list of numbers corresponding to number ranges in a string"""
    g = (
        (l,h)
        for r in ranges.split(',')
        for l,h in r.split('-')
        )
    return (x
            for range_tuple in g
            for l,h in range_tuple
            for x in range(int(l), int(h)+1)
            )

parse_ranges('0-5,8-10,11-13')

Answer 1

如果您的目标是将该格式的字符串解析为int列表，那么使用正则表达式来解析它可能是值得的

def parse_ranges(ranges):
    """Return a list of numbers corresponding to number ranges in a string"""
    return re.split(',|-', ranges)

parse_ranges('0-5,8-10,11-13')

输出：

['0', '5', '8', '10', '11', '13']

要将它们作为int而不是字符串获取，可以执行以下操作：

def parse_ranges(ranges):
    """Return a list of numbers corresponding to number ranges in a string"""
    return (int(x) for x in re.split(',|-', ranges))

list(parse_ranges('0-5,8-10,11-13'))

输出：

[0, 5, 8, 10, 11, 13]

Answer 2

intspan模块就是这样做的。

import intspan
s =  '0-0,3-5,7-10'

>>> list(intspan.intspan(s))
[0, 3, 4, 5, 7, 8, 9, 10]

Answer 3

你的循环太多了。因为你试图将两个数字放在一起，所以你不需要第二个嵌套循环。第二个是平坦结果：

>>> [x for r in s.split(',') for x in r.split('-')]
['0', '0', '3', '5', '7', '10']

使用其中一个保持数字成对：

>>> [r.split('-') for r in s.split(',')]
[['0', '0'], ['3', '5'], ['7', '10']]

然后，在第二次理解中，您可以将前两个循环组合在一起，方法是将l和h拆包到位。

return (x
        for l,h in g
        for x in range(int(l), int(h)+1)
        )