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

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

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


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

Задача

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

Решение

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

Пример

№1

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

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

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

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

Результат:

[1, 5, 9]

См.также

Ссылки на полезные ресурсы

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