Python：np.tril_indices()

获取方阵下三角所有元素的索引 

import numpy as np

L = np.zeros((5,5))
A = np.tril_indices(5)
print(A)
L[np.tril_indices(5)] = 1
print(L)

(array([0, 1, 1, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 4], dtype=int64),

array([0, 0, 1, 0, 1, 2, 0, 1, 2, 3, 0, 1, 2, 3, 4], dtype=int64))
[[1. 0. 0. 0. 0.]
 [1. 1. 0. 0. 0.]
 [1. 1. 1. 0. 0.]
 [1. 1. 1. 1. 0.]
 [1. 1. 1. 1. 1.]]