numpy.triu_indices

numpy.triu_indices(n, k=0, m=None) [source]

Возвращает индексы для верхней треугольной части массива (n, m).

Параметры:	n : int Размер массивов, для которых возвращаемые индексы будут валидны. k : int, необязательно Смещение диагонали (см. triu для подробностей). m : int, необязательно Новое в версии 1.9.0. Размер столбцов массивов, для которых возвращаемые массивы будут валидны. По умолчанию m принимается равным n.
Возвращает:	inds : кортеж, форма (2) из ndarrays, форма (n) Индексы для треугольной части. Возвращаемый кортеж содержит два массива, каждый с индексами вдоль одной размерности массива. Может быть использован для среза ndarray формы (n, n).

См. также

tril_indices

аналогичная функция, для нижней треугольной части.

mask_indices

обобщенная функция, принимающая произвольную функцию маскирования.

triu, tril

Примечания

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

Примеры

Вычислите два набора индексов для доступа к массивам 4x4, один для верхней треугольной части, начинающейся с главной диагонали, а другой — начиная с диагонали на две позиции правее:

>>> iu1 = np.triu_indices(4)
>>> iu2 = np.triu_indices(4, 2)

Вот как они могут быть использованы с образцовым массивом:

>>> 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]])

Оба для индексирования:

>>> a[iu1]
array([ 0,  1,  2,  3,  5,  6,  7, 10, 11, 15])

И для присваивания значений:

>>> a[iu1] = -1
>>> a
array([[-1, -1, -1, -1],
       [ 4, -1, -1, -1],
       [ 8,  9, -1, -1],
       [12, 13, 14, -1]])

Эти индексы охватывают только небольшую часть всего массива (две диагонали правее главной):

>>> a[iu2] = -10
>>> a
array([[ -1,  -1, -10, -10],
       [  4,  -1,  -1, -10],
       [  8,   9,  -1,  -1],
       [ 12,  13,  14,  -1]])