np.triu_indices_from() 

返回方阵的上三角矩阵的索引

numpy.triu_indices_from（arr，k = 0 ）

参数：

arr ：ndarray，shape（N，N）

k： int，可选，对角线偏移

例子：

import numpy as np

a = np.array([[1,2],[4,5]])
np.triu_indices_from(a)
#(array([0, 0, 1], dtype=int64), array([0, 1, 1], dtype=int64))

https://numpy.org/doc/stable/reference/generated/numpy.triu_indices_from.html