Получить разные столбцы в каждом ряду

Я хотел бы получить разные столбцы из каждой строки из матрицы. Например:

A = [1,2;1,4;5,2]
B = [2;2;1]

выход должен дать:

out = [2;4;5]

Итак, вкратце:A это матрица иB имеет индексы для столбцов в ряду. Как я могу сделать это без использования цикла (если этовозможно)?

Ответы на вопрос(2)

Ваш ответ на вопрос