python - 获取具有给定索引的 Python 列表的子列表？

Question

我有一个 Python 列表，比如 a = [0,1,2,3,4,5,6]。我还有一个索引列表，比如 b = [0,2,4,5]。如何在 b 中获取带有索引的 a 元素列表？

Answer 1

您可以使用 list comprehension 来获取该列表:

c = [a[index] for index in b]
print c

这相当于:

c= []
for index in b:
    c.append(a[index])
print c

输出:

[0,2,4,5]

注意:

请记住，some_list[index] 是用于访问特定索引中 list 元素的符号。