numpy.tril_indices_from

numpy.tril_indices_from(arr, k=0)

Функция tril_indices_from() возвращает индексы элементов нижнего треугольника указанного массива.

Параметры:
arr - массив NumPy.
Двумерный массив NumPy.
k - целое число.
Определяет смещение диагонали треугольника.
Возвращает:
tuple - кортеж массивов NumPy
кортеж массивов с индексами элементов нижнего треугольника указанного массива.
Смотрите так же:
tril_indices, triu_indices, triu_indices_from, tril, triu


Примеры

Данная функция абсолютно аналогична функции tril_indices(), только вместо параметров с размерами массива, она сразу принимает указанный массив:

>>> import numpy as np
>>> 
>>> a = np.arange(20).reshape(5, 4)
>>> a
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15],
       [16, 17, 18, 19]])
>>> 
>>> np.tril_indices_from(a)
(array([0, 1, 1, ..., 4, 4, 4], dtype=int32), array([0, 0, 1, ..., 1, 2, 3], dtype=int32))
>>> 
>>> a[np.tril_indices_from(a)] = 77
>>> a
array([[77,  1,  2,  3],
       [77, 77,  6,  7],
       [77, 77, 77, 11],
       [77, 77, 77, 77],
       [77, 77, 77, 77]])

Параметр k, позволяет указать смещение диагонали треугольника:

>>> a[np.tril_indices_from(a, k = -2)] = 0
>>> a
array([[77,  1,  2,  3],
       [77, 77,  6,  7],
       [ 0, 77, 77, 11],
       [ 0,  0, 77, 77],
       [ 0,  0,  0, 77]])