python: извлечение подматриц из индексов

Question

У меня есть матрица и список индексов для обоих измерений.

A = [[1,2,3],[4,5,6],[7,8,9]]
idx = [1,3,4]
idy = [2,5,7]

Что такое похожая запись от Matlab, делающего A (idx, idy) в Python?

Answer 1

Вот мой пример:

A = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

idx = [0, 1]
idy = [1, 2]

a = [[A[ix][iy] for iy in idy] for ix in idx]
print(a)

От:

[
   [1, 2, 3], 
   [4, 5, 6], 
   [7, 8, 9]
]

Я получаю подматрицу по моим индексам idx и idy:

[
    [2, 3], 
    [5, 6]
]

Имейте в виду, что в python индексация начинается с 0, поэтому, учитывая, что здесь мы имеем матрицу 3x3, наименьший индекс равен 0, а самый большой - 2 для каждой строки и столбца.

Это то, что вы ищете? Пожалуйста, дайте мне знать.