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

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

你的循环太多了。由于您试图将两个数字保持在一起，因此不需要第二个嵌套循环。第二个是压平结果:

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

使用 1 来保持数字配对:

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

然后，通过第二个推导式，您可以通过将 l 和 h 解包来组合前两个 for 循环。

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