numpy.tril_indices

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

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

Параметры:	n : int Размерность строк массивов, для которых возвращаемые индексы будут валидны. k : int, необязательно Смещение диагонали (см. tril для подробностей). m : int, необязательно Новое в версии 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,  8,  9, 10, 12, 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]])