📜  Python中的numpy.tri

📅  最后修改于: 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.]]