numpy.triu_indices

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

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

Параметры

nint

Размер массивов, для которых возвращаемые индексы будут верны.

kint, optional

Смещение диагонали (см. triu для подробностей).

mint, optional

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

Размер столбца массивов, для которых возвращаемые массивы будут верны. По умолчанию m принимается равным n.

Возвращает

indstuple, shape(2) of ndarrays, shape(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, ..., 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]])