Получить угловые значения в Python NumPy ndarray

Я пытаюсь получить доступ к угловым значениям NumPy ndarray. Я'м абсолютно тупик, как по методологии. Любая помощь будет принята с благодарностью.

Например, из приведенного ниже массива I 'Мне нравится возвращаемое значение массива ([1,0,0,5]) или массива ([[1,0], [0,5]]).

array([[ 1.,  0.,  0.,  0.],
       [ 0.,  1.,  0.,  0.],
       [ 0.,  0.,  1.,  5.],
       [ 0.,  0.,  5.,  5.]])

Ответы на вопрос(4)

Ваш ответ на вопрос