python-使用索引的嵌套列表切片嵌套列表

Question

我有一个嵌套列表

import numpy as np
mylist = np.array([[12, 11, 14, 15, 66],
                   [3, 5, 6],
                   [13, 4, 7, 33, 98]])
myindex = [[0, 2, 3],
           [2],
           [1, 2, 3, 4]]

如何使用mylist切片myindex？

非常感谢。

Answer 1

您可以使用zip和列表理解功能。

slices = []
for ix, l in zip(myindex, mylist): 
    slices.append([l[i] for i in ix])

输出

[[12, 14, 15], [6], [4, 7, 33, 98]]