Диагональ в NumPy

Функция diag для двумерной таблицы возвращает одномерный ряд из чисел на главной диагонали: от левого верхнего угла к правому нижнему.

Снимем диагональ с квадратной таблицы:

import numpy as np

square = np.array([[5, 6], [7, 8]])
print(np.diag(square)) # выведет [5 8]

Та же функция умеет собрать диагональную таблицу из одномерного ряда: на главной линии стоят элементы ряда, остальные ячейки нулевые:

import numpy as np

values = np.array([5, 8])
print(np.diag(values)) # выведет [[5 0]
                        #           [0 8]]

Задайте таблицу [[9, 1, 2], [3, 4, 5], [6, 7, 8]]. Выведите ряд чисел с главной диагонали.

Создайте квадратную таблицу 3 на 3 с разными целыми значениями. Выведите одномерный ряд с её главной диагонали.