np.triu()和np.tril()的用法介绍
作者:互联网
tri:triangle,三角。
u:upper,上三角
l:lower,下三角
所以,前者是返回原矩阵的上三角矩阵,后者是下三角矩阵。看招:
a=np.array([[1,2],[3,4]])
print(np.tril(a))
print(np.triu(a))
可以看到,保留对应的部分,其他清空为0.
值得注意的是,这个不一定只适用于方针,例如:
a=np.array([[1,2],[3,4],[5,6]])
print(np.tril(a))
print(np.triu(a))
此时我们可以使用这两个函数的另外一个参数k,以np.tril为例,如下:
默认k=0。
以k=1为例,其表示主对角线上面的第一个对角线的上面全部清为0,在我们这里,主对角线是(1,4),上一个对角线是(2),再往上面没有了元素,所以清0相当于没有清0,如下:
a=np.array([[1,2],[3,4],[5,6]])
print(np.tril(a,k=1))
标签:tril,三角,print,triu,对角线,np 来源: https://blog.csdn.net/qq_43391414/article/details/120957737