🔡 Транспонирование матриц Python.

Допустим есть матрица (двумерный список) размера 2x3:

>>> matrix = [[1, 2, 3], [4, 5, 6]]

Как в одну строку кода можно поменять местами строки и столбцы (научно: транспонировать матрицу)?

>>> list(zip(*matrix))

[(1, 4), (2, 5), (3, 6)]

Объяснение. Звездочка перед значением аргумента функции разворачивает список в список аргументов. Пример:

>>> print(*matrix, sep=', ')

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

В списке matrix было 2 элемента, и вызов print стал аналогичен:

>>> print(matrix[0], matrix[1], sep=', ')

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

Теперь zip. Она берет на вход последовательности (одну, две, больше...) и вовращает список из кортежей вида:

[ (элем0_из_первой_посл, элем0_из_второй_посл, ...),

 (элем1_из_первой_посл, элем1_из_второй_посл, ...),

 ... ]

У нас будет:

[ (matrix[0][0] = 1, matrix[1][0] = 4),

 (matrix[1][0] = 2, matrix[1][1] = 5),

 (matrix[2][0] = 3, matrix[2][1] = 6) ]

Но нам нужен не список кортежей, а список списков – резонно возразите вы. Что ж! Применим к каждому элементу результата ф-цию list, она превратит его в список:

>>> list(map(list, zip(*matrix)))

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

Это и есть ответ. Кстати, размер матрицы стал 3x2. 

А потом добавьте, что вы бы не страдали велосипедо-строением для транспонирования матриц и применили бы numpy:

>>> matrix = numpy.array([[1, 2, 3], [4, 5, 6]])

>>> matrix.T

array([[1, 4],

    [2, 5],

    [3, 6]])

#machinelearning #artificialintelligence #ai #datascience #python #programming #technology #deeplearning