numpy.triu_indices_from

numpy.triu_indices_from(arr, k=0)[source]

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

См. triu_indices для подробных сведений.

Параметры:

arrndarray, форма(N, N)

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

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

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

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

triu_indices_fromкортеж, форма(2) от ndarray, форма(N)

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

См. также

triu_indices, triu, tril_indices_from

Примечания

Введено в версии 1.4.0.

Примеры

Создайте массив 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]])

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

>>> triui = np.triu_indices_from(a)
>>> triui
(array([0, 0, 0, 0, 1, 1, 1, 2, 2, 3]), array([0, 1, 2, 3, 1, 2, 3, 2, 3, 3]))

>>> a[triui]
array([ 0,  1,  2,  3,  5,  6,  7, 10, 11, 15])

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

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

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

>>> triuim1 = np.triu_indices_from(a, k=1)
>>> a[triuim1]
array([ 1,  2,  3,  6,  7, 11])