numpy.tril_indices

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

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

Параметры

nint

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

kint, необязательно

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

mint, необязательно

Добавлен в версии 1.9.0.

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

Возвращает

indsкортеж массивов

Индексы для треугольника. Возвращаемый кортеж содержит два массива, каждый из которых содержит индексы по одному измерению массива.

См. также

triu_indices

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

mask_indices

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

tril, triu

Примечания

Добавлен в версии 1.4.0.

Примеры

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

>>> il1 = np.tril_indices(4)
>>> il2 = np.tril_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[il1]
array([ 0,  4,  5, ..., 13, 14, 15])

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

>>> a[il1] = -1
>>> a
array([[-1,  1,  2,  3],
       [-1, -1,  6,  7],
       [-1, -1, -1, 11],
       [-1, -1, -1, -1]])

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

>>> a[il2] = -10
>>> a
array([[-10, -10, -10,   3],
       [-10, -10, -10, -10],
       [-10, -10, -10, -10],
       [-10, -10, -10, -10]])