Сортировать 2D зубчатый список на основе пользовательского порядка строк, используя Python 2

Question

Предположим, у меня есть следующий зубчатый список в Python:

a = [[73, 74, 93, 97, 147],
     [5, 11, 12, 47],
     [987, 988, 989, 990, 991],
     [15, 16, 17, 22, 71, 988],
     [15, 16, 17]]

У меня также есть следующий список:

b = [5, 3, 2, 4, 1]

Я хочу отсортировать строки списка a на основе порядка сортировки, предоставленного b, т.е. после сортировки a должно быть:

a = [[15, 16, 17],
     [987, 988, 989, 990, 991],
     [5, 11, 12, 47],
     [15, 16, 17, 22, 71, 988],
     [73, 74, 93, 97, 147]]

Я не понимаю, как это сделать. Пожалуйста, помогите мне.

Answer 1

Вы можете сделать это так:

output = [a[j-1] for j in b]