NumPy: извлечь верхнюю треугольную часть матрицы

NumPy: Array Object Exercise-153 с решением

Напишите программу NumPy для извлечения верхней треугольной части матрицы NumPy.

Пример решения :

Код Python:

import numpy as np
num = np.arange(18)
arr1 = np.reshape(num, [6, 3])
print("Original array:")
print(arr1)
result  = arr1[np.triu_indices(3)]
print("\nExtract upper triangular part of the said array:")
print(result)
result  = arr1[np.triu_indices(2)]
print("\nExtract upper triangular part of the said array:")
print(result)

Пример вывода:

 Исходный массив:
[[0 1 2]
 [3 4 5]
 [6 7 8]
 [9 10 11]
 [12 13 14]
 [15 16 17]]
Извлеките верхнюю треугольную часть указанного массива:
[0 1 2 4 5 8]
Извлеките верхнюю треугольную часть указанного массива:
[0 1 4]

Иллюстрированная презентация:

Редактор кода Python:

Есть другой способ решить это решение? Внесите свой код (и комментарии) через Disqus.

Предыдущий: Напишите программу NumPy для расчета суммы всех столбцов двумерного массива.
Далее: Напишите программу NumPy, чтобы получить копию матрицы с обнуленными элементами ниже k-й диагонали.