📅  最后修改于: 2020-06-04 14:57:20             🧑  作者: Mango
numpy.tri(R, C = None, k = 0, dtype = ‘float’) : 创建一个数组,该数组在给定对角线(约k)处和以下(约k)处为1,在其他位置为0。
参数:
R:行数
C:[可选]列数;默认情况下R = C
k:[int,可选,默认为0]
我们需要对角线; k> 0表示对角线高于主对角线,反之亦然。
dtype:[可选,float(byDefault)]返回的数组的数据类型。
# Python程序说明numpy.tri方法
import numpy as geek
print("tri with k = 1 : \n",geek.tri(2, 3, 1, dtype = float), "\n")
print("tri与主对角线 : \n",geek.tri(3, 5, 0), "\n")
print("tri with k = -1 : \n",geek.tri(3, 5, -1), "\n")
输出:
tri with k = 1:
[[1. 1. 0.]
[1. 1. 1.]]
tri与主对角线:
[[1. 0. 0. 0. 0.]
[1. 1. 0. 0 。0.]
[1. 1. 1. 0. 0.]]
tri with k = -1:
[[0. 0. 0. 0. 0.]
[1. 0. 0. 0. 0.]
[ 1. 1. 0. 0. 0.]]