import numpy as np
n_class = 5
L = np.zeros((n_class - 1, n_class - 1))
print(L)
L[np.tril_indices(n_class-1)] = 1.
print(L)

本文介绍了一种使用Python的NumPy库创建特定形状的三角矩阵的方法。通过设置矩阵大小并利用NumPy的tril_indices函数,可以方便地生成一个下三角矩阵,并将非零元素设为1。
import numpy as np
n_class = 5
L = np.zeros((n_class - 1, n_class - 1))
print(L)
L[np.tril_indices(n_class-1)] = 1.
print(L)

6237
766

被折叠的 条评论
为什么被折叠?
&spm=1001.2101.3001.5002&articleId=115918644&d=1&t=3&u=360534cf480145c19d1fe39e4c7ec084)