numpy.tril_indices_from

numpy.tril_indices_from(arr, k=0)

Возвращает индексы для нижнего треугольника массива arr.

См. tril_indices для полной информации.

Параметры:

arrarray_like

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

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

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

Смотрите также

tril_indices, tril, triu_indices_from

Примеры

Попробуйте в вашем браузере!

>>> import numpy as np

Создать массив 4 на 4

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

Передайте массив, чтобы получить индексы элементов нижнего треугольника.

>>> trili = np.tril_indices_from(a)
>>> trili
(array([0, 1, 1, 2, 2, 2, 3, 3, 3, 3]), array([0, 0, 1, 0, 1, 2, 0, 1, 2, 3]))

>>> a[trili]
array([ 0,  4,  5,  8,  9, 10, 12, 13, 14, 15])

Это синтаксический сахар для tril_indices().

>>> np.tril_indices(a.shape[0])
(array([0, 1, 1, 2, 2, 2, 3, 3, 3, 3]), array([0, 0, 1, 0, 1, 2, 0, 1, 2, 3]))

Используйте k параметр для возврата индексов нижней треугольной матрицы до k-й диагонали.

>>> trili1 = np.tril_indices_from(a, k=1)
>>> a[trili1]
array([ 0,  1,  4,  5,  6,  8,  9, 10, 11, 12, 13, 14, 15])

НазадОткрыть во вкладке