Python:Рецепты/Получить диагональ матрицы

Материал из Онлайн справочника
Перейти к навигацииПерейти к поиску

Проверка/Оформление/Редактирование: Мякишев Е.А.


Получить диагональ матрицы

Задача

Получить диагональ матрицы

Решение

Для этого можно воспользоваться выражением спискового включения.

Пример

№1

Дана матрица:

M = [[1,2,3],
     [4,5,6],
     [7,8,9]]

Воспользуемся выражением спискового включения и получим диагональ матрицы:

[M[i][i] for i in [0,1,2]]

Результат:

[1, 5, 9]

См.также

Внешние ссылки