8

我有一个看起来像这样的列表:

lst = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]

我想生成一个看起来像这样的过滤列表:

filtered_lst = [2, 6, 7, 9, 10, 13]

Python 是否提供自定义切片的约定。例如:

lst[1, 5, 6, 8, 9, 12] # slice a list by index
4

5 回答 5

16

使用operator.itemgetter()

from operator import itemgetter

itemgetter(1, 5, 6, 8, 9, 12)(lst)

演示:

>>> from operator import itemgetter
>>> lst = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]
>>> itemgetter(1, 5, 6, 8, 9, 12)(lst)
(2, 6, 7, 9, 10, 13)

这将返回一个元组;强制转换为带有list(itemgetter(...)(lst))if a 的列表。

请注意,这等效于lst[start:stop]具有一组索引而不是范围的切片表达式 ( );它不能用作左侧切片赋值 ( lst[start:stop] = some_iterable)。

于 2013-01-01T22:01:20.657 回答
5

Numpy 数组具有这种切片语法:

In [45]: import numpy as np

In [46]: lst = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13])

In [47]: lst[[1, 5, 6, 8, 9, 12]]
Out[47]: array([ 2,  6,  7,  9, 10, 13])
于 2013-01-01T22:08:13.627 回答
1

我会采用operator.itemgetter()Martijn Pieters 建议的方法,但这是另一种方法(为了完整性)

In [23]: lst = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]

In [24]: indices = set([1, 5, 6, 8, 9, 12])

In [25]: [n for i,n in enumerate(lst) if i in indices]
Out[25]: [2, 6, 7, 9, 10, 13]
于 2013-01-01T22:05:58.950 回答
1

使用列表推导可以轻松直接地完成。

lst = range(1, 14)
indices = [1, 5, 6, 8, 9, 12]
filtered_lst = [lst[i] for i in indices]
于 2013-01-01T22:20:47.133 回答
0

Python 切片允许您将切片作为赋值的目标。并且 Python 切片语法不允许具有不规则索引模式的切片。所以,如果你想让你的“自定义”切片成为赋值的目标,那么 Python 切片语法是不可能的。

如果通过获取指定元素的副本来满足您的要求,则operator.itemgetter()满足您的需求。如果您需要切片分配,那么 numpy 切片是一个不错的选择。

于 2013-01-01T22:12:23.270 回答