Курс Python → Создание матрицы в Python

В Python есть возможность использовать вложенные list comprehensions для создания матрицы, то есть списка списков. Это означает, что вы можете использовать list comprehension не только для создания простого списка, но и для создания списка, который содержит другие списки внутри себя. Такой подход часто используется при работе с матрицами, таблицами или другими структурами данных, где требуется хранить данные в виде двумерного массива.

Для создания вложенных list comprehensions вам нужно просто добавить еще один цикл внутрь уже существующего выражения. Таким образом, вы можете создать матрицу, указав два цикла — один для итерации по строкам, другой для итерации по столбцам. В результате получится список списков, где каждый внутренний список представляет собой строку матрицы.


matrix = [[i*j for j in range(1, 4)] for i in range(1, 4)]
print(matrix)

В приведенном примере мы создаем матрицу 3×3, где каждый элемент равен произведению индексов строки и столбца. При этом внешний цикл отвечает за итерацию по строкам (от 1 до 3), а внутренний цикл — по столбцам (от 1 до 3). Результатом выполнения этого кода будет следующая матрица:


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

Таким образом, использование вложенных list comprehensions позволяет компактно и элегантно создавать матрицы и другие структуры данных в Python, что упрощает работу с данными и улучшает читаемость кода. Помимо умножения элементов, вы можете использовать любые другие выражения и операции внутри вложенных list comprehensions для генерации нужной структуры данных.

Твои коллеги будут рады, поделись в