Матричная транспонирование NumPy – Транспонирование массива в Python

Транспонирование матрицы получается путем перемещения данных строк в столбцы и данных столбцов в строки. Если у нас есть массив формы (X, Y), то транспонированный массив будет иметь форму (Y, X).

Транспонирование матрицы в NumPy

Модуль Python numpy в основном используется для работы с массивами в Python. Мы можем использовать функцию transpose(), чтобы получить транспонированный массив.

import numpy as np

arr1 = np.array([[1, 2, 3], [4, 5, 6]])

print(f'Original Array:\n{arr1}')

arr1_transpose = arr1.transpose()

print(f'Transposed Array:\n{arr1_transpose}')

Вывод:

Original Array:
[[1 2 3]
 [4 5 6]]
Transposed Array:
[[1 4]
 [2 5]
 [3 6]]

Транспонирование объекта подобного массива

Функция transpose() также работает с объектом, подобным массиву, таким как вложенный список.

arr1 = [[1, 2, 3], [4, 5, 6]]

arr1_transpose = np.transpose(arr1)

Результат будет таким же, как и в предыдущей программе. Ссылка: Документация по API