如何引用列表中不同范围的值？

Question

我想做的是从一个列表中引用几个不同的范围，即。我想要4-6元素，12-18元素，等等。这是我最初的尝试：

test = theList[4:7, 12:18]

我希望它能做同样的事情：

test = theList[4,5,6,12,13,14,15,16,17]

但是我有一个语法错误。最好/最简单的方法是什么？

Answer 1

您可以添加@Bhargav_Rao所述的两个列表。更一般地说，您还可以使用列表生成器语法：

test = [theList[i] for i in range(len(theList)) if 4 <= i <= 7 or 12 <= i <= 18]

Answer 2

您还可以使用itertools模块：

>>> from itertools import islice,chain
>>> theList=range(20)
>>> list(chain.from_iterable(islice(theList,*t) for t in [(4,7),(12,18)]))
[4, 5, 6, 12, 13, 14, 15, 16, 17] 

请注意，因为islice在每次迭代中返回生成器，所以在内存使用方面，它比列表切片执行得更好。

此外，您可以使用一个函数为更多的索引和一般的方式。

>>> def slicer(iterable,*args):
...    return chain.from_iterable(islice(iterable,*i) for i in args)
... 
>>> list(slicer(range(40),(2,8),(10,16),(30,38)))
[2, 3, 4, 5, 6, 7, 10, 11, 12, 13, 14, 15, 30, 31, 32, 33, 34, 35, 36, 37]

注意：如果要循环结果，不需要将结果转换为列表！

Answer 3

您可以添加这两个列表。

>>> theList = list(range(20))
>>> theList[4:7] + theList[12:18]
[4, 5, 6, 12, 13, 14, 15, 16, 17]