9

在我问之前，我做了一些谷歌搜索，但无法找到答案。

我的场景是这样的：一个数字列表被传递给脚本，要么通过文件以 \n 分隔，要么通过命令行 arg 以逗号分隔。这些数字可以是单数，也可以是块，如下所示：

文件：

1
2
3
7-10
15
20-25

命令行参数：

1, 2, 3, 7-10, 15, 20-25

两者最终都在同一个列表[]中。我想扩展 7-10 或 20-25 块（显然在实际脚本中这些数字会有所不同）并将它们附加到一个新列表中，最终列表如下所示：

['1','2','3','7','8','9','10','15','20','21','22','23','24','25']

我知道像 .append(range(7,10)) 这样的东西可以在这里帮助我，但我似乎无法找出原始 list[] 的哪些元素需要扩展。

所以，我的问题是：给定一个列表[]：

['1','2','3','7-10','15','20-25'],

我怎样才能得到一个列表[]：

 ['1','2','3','7','8','9','10','15','20','21','22','23','24','25']
4

3 回答 3

14

所以假设你得到了这个列表：

L = ['1','2','3','7-10','15','20-25']

并且您想扩展其中包含的所有范围：

answer = []
for elem in L:
    if '-' not in elem:
        answer.append(elem)
        continue
    start, end = elem.split('-')
    answer.extend(map(str, range(int(start), int(end)+1)))

当然，有一个方便的单线：

answer = list(itertools.chain.from_iterable([[e] if '-' not in e else map(str, range(*[int(i) for i in e.split('-')]) + [int(i)]) for e in L]))

但这利用了 python2.7 中泄漏变量的性质，我认为这在 python3 中不起作用。此外，它并不是最易读的代码行。所以我不会真的在生产中使用它，如果我是你……除非你真的讨厌你的经理。

参考：  append()  continue  split()  extend()  map()  range()  list()  itertools.chain.from_iterable()  int()

于 2015-05-12T20:52:36.213 回答
2

输入：

arg = ['1','2','3','7-10','15','20-25']

输出：

out = []
for s in arg:
    a, b, *_ = map(int, s.split('-') * 2)
    out.extend(map(str, range(a, b+1)))

或者（在 Python 2 中）：

out = []
for s in arg:
    r = map(int, s.split('-'))
    out.extend(map(str, range(r[0], r[-1]+1)))
于 2015-05-12T21:06:51.613 回答
0

好的旧 map + reduce 会派上用场：

>>> elements = ['1','2','3','7-10','15','20-25']
>>> reduce(lambda original_list, element_list: original_list + map(str, element_list), [[element] if '-' not in element else range(*map(int, element.split('-'))) for element in elements])
['1', '2', '3', '7', '8', '9', '15', '20', '21', '22', '23', '24']

好吧，除了您希望 20-25 也包含 25 之外，这将起到作用......所以这里有更多的汤：

reduce(
    lambda original_list, element_list: original_list + map(str, element_list), 
    [[element] if '-' not in element 
     else range(int(element.split('-')[0]), int(element.split('-')[1]) + 1) 
     for element in elements])

现在，即使这可行，您也可能最好使用一些 for 循环。这就是他们在 python 3 中删除 reduce 的原因。

于 2015-05-12T23:15:31.737 回答