Эффективные операции ndarray

Question

Я конвертирую некоторый код Matlab в Python. Мне нужно сделать некоторые манипуляции с матрицами. Моя матрица (A) есть (прямо сейчас) матрица 65x3. Однако количество строк зависит от того, на каком этапе я участвую в программе.

В Matlab, код, я работаю на это:

output = inv(A'*A) * A'; 

Следующий код Python воспроизводит ожидаемый выход только штрафом. Мне просто интересно, есть ли способ сделать это лучше (более Pythonic, быстрее и т. Д.)? Я пытаюсь придерживаться только базового Python и numpy.

output = np.dot(np.linalg.inv(np.dot(np.transpose(A), A)), np.transpose(A)) 

Спасибо всем, кто готов помочь.

Answer 1

Вы можете использовать атрибут T (переносит массив). Кроме того, при использовании Python 3.5 вы можете использовать @ для точечного продукта (см. PEP 465).

output = np.linalg.inv(A.T @ A) @ A.T