Python:Рецепты/Получить диагональ матрицы
Материал из Онлайн справочника
Перейти к навигацииПерейти к поиску
Проверка/Оформление/Редактирование: Мякишев Е.А.
Получить диагональ матрицы
Задача
Получить диагональ матрицы
Решение
Для этого можно воспользоваться выражением спискового включения.
Пример
№1
Дана матрица:
M = [[1,2,3],
[4,5,6],
[7,8,9]]
Воспользуемся выражением спискового включения и получим диагональ матрицы:
[M[i][i] for i in [0,1,2]]
Результат:
[1, 5, 9]