numpy.diag_indices_from

numpy.diag_indices_from(arr)[source]

Возвращает индексы для доступа к главной диагонали n-мерного массива.

См. diag_indices для получения подробной информации.

Параметры:

arrмассив, по крайней мере 2-мерный

См. также

diag_indices

Примечания

Новая в версии 1.4.0.

Примеры

Создайте массив 4 на 4.

>>> a = np.arange(16).reshape(4, 4)
>>> a
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15]])

Получите индексы элементов диагонали.

>>> di = np.diag_indices_from(a)
>>> di
(array([0, 1, 2, 3]), array([0, 1, 2, 3]))

>>> a[di]
array([ 0,  5, 10, 15])

Это просто синтаксический сахар для diag_indices.

>>> np.diag_indices(a.shape[0])
(array([0, 1, 2, 3]), array([0, 1, 2, 3]))