python保留矩阵的上对角

您好，关于保留矩阵的上对角，可以使用numpy库中的triu函数来实现。具体操作如下：

import numpy as np

a = np.array([[1,2,3],[4,5,6],[7,8,9]])

b = np.triu(a)

print(b)

输出结果为：

[[1 2 3]

[0 5 6]

[0 0 9]]

其中，triu函数将矩阵a的上三角部分保留，下三角部分设为0，得到了新的矩阵b。