我有一个函数,它采用'1,3-5,7,19'等格式的字符串,并将输出列表[1, 3, 4, 5, 7, 19]
。使用if/else的嵌套列表理解
但是,我在想这也许足够简单,可以用嵌套的列表理解。
我原来的功能是:
result = []
for x in row_range.split(','):
if '-' in x:
for y in range(int(x.split('-')[0]), int(x.split('-')[1]) + 1)):
result.append(y)
else:
result.append(int(x))
我认为理解是这样的:
result = [y for x in row_range.split(',') if '-' in x else int(x) for y in range(int(x.split('-')[0]), int(x.split('-')[1] + 1)]
甚至
result = [y for x in row_range.split(',') if '-' in x for y in range(int(x.split('-')[0]), int(x.split('-')[1] + 1) else int(x)]
但这些都是SyntaxError错误。将if/else移到理解的前面
result = [y if '-' in x else int(x) for x in row_range.split(',') for y in range(int(x.split('-')[0]), int(x.split('-')[1]) + 1)]
导致IndexError:列表索引超出范围。
这可能吗?我已经有一个很好地处理它的函数,并且更具可读性,但是我只是好奇这是否可以在python中完成。
只是为了好玩,我决定[在Haskell实现这个(HTTPS://gist.github。 com/NotTheEconomist/e47de2ee01ae3fcde1926d9df705a079)(使用'split'和'text'包) –