numpy.tril_indices

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

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

Параметры

nint

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

kint, optional

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

mint, optional

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

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

Возвращаемое значение

indstuple of arrays

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

См. также

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