У меня есть матрица A размера 256 X 16 и вторая матрица B размера 8 X 8, где каждый элемент B является целым числом в диапазоне [0, 15]. Если бы A был просто вектором строки (т.е. размером 1X16), то C = A[B] вывел бы матрицу 8X8, где Cij = A[bij]. Есть ли какой-нибудь эффективный способ сделать то же самое, если A - 256X16, используя numpy (или что-нибудь еще)? Я имею в виду, что мне нужна выходная матрица C размером 256 X 8 X 8 (т. Е. B индексирует каждую строку A). Спасибо!