# 通过numpy.triu产生上三角矩阵
>>> import numpy as np
>>> np.triu([[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15],[16,17,18,19,20]])
array([[ 1, 2, 3, 4, 5],
[ 0, 7, 8, 9, 10],
[ 0, 0, 13, 14, 15],
[ 0, 0, 0, 19, 20]])
>>> np.triu([[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15],[16,17,18,19,20]],k=-1)
array([[ 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10],
[ 0, 12, 13, 14, 15],
[ 0, 0, 18, 19, 20]])
>>> np.triu([[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15],[16,17,18,19,20]],k=0)
array([[ 1, 2, 3, 4, 5],
[ 0, 7, 8, 9, 10],
[ 0, 0, 13, 14, 15],
[ 0, 0, 0, 19, 20]])
>>> np.triu([[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15],[16,17,18,19,20]],k=1)
array([[ 0, 2, 3, 4, 5],
[ 0, 0, 8, 9, 10],
[ 0, 0, 0, 14, 15],
[ 0, 0, 0, 0, 20]])
>>> np.triu([[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15],[16,17,18,19,20]],k=2)
array([[ 0, 0, 3, 4, 5],
[ 0, 0, 0, 9, 10],
[ 0, 0, 0, 0, 15],
[ 0, 0, 0, 0, 0]])
>>> np.triu([[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15],[16,17,18,19,20]],k=3)
array([[ 0, 0, 0, 4, 5],
[ 0, 0, 0, 0, 10],
[ 0, 0, 0, 0, 0],
[ 0, 0, 0, 0, 0]])
numpy.triu
猜你喜欢
转载自blog.csdn.net/qq_41128383/article/details/105668731
今日推荐
周排行