问题描述:
取python 列表中的一些下标所对应的元素,这些下标不是连续的,也不具有一定的规律
例:
l = [0,1,2,3,4,5,6,7,8,9,10]
index = [2,6,3,10]
- for 循环 (最容易想到的方式)
res = []
for i in index:
res.append(l[i])
- 列表表达式
res = [l[i] for i in index]
- numpy数组
import numpy as np
l_np = np.array(l)
res = l_np[index]
- operator
from operator import itemgetter
res = itemgetter(*index)(l)
参考:
https://stackoverflow.com/questions/14114621/complex-list-slice-index-in-python