通过索引列表从列表中删除元素

Question

我有一个索引列表，

[[ 0.2, 0.60000002], [ 0.40000001, 1.20000005]]

另一份我的数据清单：

ind = [0,1,2,5,6,7]

我希望仅保留与data = [0,1,4,9,16,25,36,49,64] 中的值对应的data索引：

list

或重新编辑，删除result = [0,1,4,25,36,49] 中不属于data值的索引。

我认为我可以做一些列表理解，但我无法弄清楚！谢谢，

Answer 1

您不需要删除任何内容，只需使用列表推导来使用data

中的索引从ind索引出来

>>> result = [data[i] for i in ind]
>>> result
[0, 1, 4, 25, 36, 49]