坎康的博客

原创 numpy中tril和triu的使用效果

import numpy as np A = np.array([[1,2,3],[2,3,2],[1,2,1]]) print(np.triu(A,-1)) print(np.tril(A,-1)) [[1 2 3] [2 3 2] [0 2 1]] [[0 0 0] [2 0 0] [1 2 0]]