Python NumPy tri方法用法及代码示例

Numpy 的 tri(~) 方法创建一个表示下三角矩阵的 2D Numpy 数组。主对角线及其下方的值用 1 填充，而其他地方用 0 填充。

1. N | int

结果数组的行数。

2. M | int | optional

结果数组的列数。默认情况下，M=N 。

3. k | int | optional

要排除或包含的对角线的数量。

k 的正值表示包含。 k=1 意味着我们在主对角线之上添加了一条附加对角线。

k 的负值表示排除。 k=-1 表示排除主对角线。 k=-2表示排除主对角线和下面的对角线。

默认情况下， k=0 ，这意味着返回一个完美的下三角形。

4. dtype | string 或 type | optional

结果数组的数据类型。默认情况下，dtype=float 。

表示下三角矩阵的 Numpy 数组。

创建 3 x 3 下三角矩阵：

np.tri(3)



array([[ 1.,  0.,  0.],
       [ 1.,  1.,  0.],
       [ 1.,  1.,  1.]])

要创建 int 类型的 3 x 4 下三角矩阵：

np.tri(3, 4, dtype=int)



array([[1, 0, 0, 0],
       [1, 1, 0, 0],
       [1, 1, 1, 0]])

要包含附加对角线，请设置 k=1 ：

np.tri(3, k=1, dtype=int)



array([[1, 1, 0],
       [1, 1, 1],
       [1, 1, 1]])

要排除主对角线，请设置 k=-1 ：

np.tri(3, k=-1)



array([[ 0.,  0.,  0.],
       [ 1.,  0.,  0.],
       [ 1.,  1.,  0.]])

相关用法

注：本文由纯净天空筛选整理自Isshin Inada大神的英文原创作品 NumPy | tri method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。