How to rearrange one list based on a second list of indices [duplicate]

Answer 1

Note: I'm assuming that m is always contiguous (i.e. there are no "gaps"), and starts at 1.

If so, you can get what you want quite easily, using a list comprehension, zip() and sorted().

Let's take it step by step:

>>> l = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']
>>> m = [1, 4, 5, 9, 2, 6, 3, 7, 8, 10]

zip() pairs each element of m with an element of l:

>>> zip(m, l)
[(1, 'a'), (4, 'b'), (5, 'c'), (9, 'd'), (2, 'e'), (6, 'f'), (3, 'g'), (7, 'h'), (8, 'i'), (10, 'j')]

sorted() returns a sorted copy of the list of pairs:

>>> sorted(zip(m, l))
[(1, 'a'), (2, 'e'), (3, 'g'), (4, 'b'), (5, 'c'), (6, 'f'), (7, 'h'), (8, 'i'), (9, 'd'), (10, 'j')]

Finally, a list comprehension takes just the second item from each pair:

>>> [x for i, x in sorted(zip(m, l))]
['a', 'e', 'g', 'b', 'c', 'f', 'h', 'i', 'd', 'j']

Answer 2

This can be done in linear O(n) time with the help of a dict:

l = ['a','b','c','d','e','f','g','h','i','j']
m = [ 1,  4,  5,  9,  2,  6,  3,  7,  8,  10]

values_by_index = dict(zip(m, l))
result = [values_by_index[index+1] for index in range(len(m))]

print(result)
# output: ['a', 'e', 'g', 'b', 'c', 'f', 'h', 'i', 'd', 'j']

---

The first line creates dict that maps indices to values, which looks like this:

>>> values_by_index
{1: 'a', 4: 'b', 5: 'c', 9: 'd', 2: 'e', 6: 'f', 3: 'g', 7: 'h', 8: 'i', 10: 'j'}

In the second line we simply loop over the indices 1 through 10 and grab the corresponding value from the dict, which is a O(1) operation in the average case.

If you have 0-based indices, simply remove the +1 from index+1.

Answer 3

If you're just trying to get elements moved around based on the other lists positions, you can loop over all elements of m and grab that element of l using list comprehension

l2 = [l[i - 1] for i in m] 

But if you do want the ordering based on the other list, you're going to need to zip them together, sort on the index, then extract the elements

[y for x,y in sorted(zip(m,l))]