Deleting elements from a list via a list of indices

Question

I have a list of indices,

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

And another list of my data:

data = [0,1,4,9,16,25,36,49,64]

I'd like to only keep the indices of data that correspond to the values in list:

result = [0,1,4,25,36,49]

Or reworded, delete the indices in data that aren't in the values of list.

I think there's some kind of list comprehension I could do but I can't figure it out! Thanks,

`list(operator.itemgetter(*ind)(data))` – Chris_Rands Jun 23 '17 at 13:37 — Chris_Rands, Jun 23 '17 at 13:37

score 3 · Answer 1 · answered Jun 23 '17 at 13:32

3

You don't need to delete anything, just use a list comprehension to index out of data using the indices from ind

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

answered Jun 23 '17 at 13:32

Cory Kramer

1 Answers1