numpy.tril_indices_from
numpy.tril_indices_from(arr, k=0)
Функция tril_indices_from() возвращает индексы элементов нижнего треугольника указанного массива.
-
- arr - массив NumPy.
- Двумерный массив NumPy.
- k - целое число.
- Определяет смещение диагонали треугольника.
-
- tuple - кортеж массивов NumPy
- кортеж массивов с индексами элементов нижнего треугольника указанного массива.
Примеры
Данная функция абсолютно аналогична функции 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]])
