numpy.tril#

numpy.tril(m, k=0)[source]#

数组的下三角部分。

返回一个数组的副本，其中第 k 条对角线以上的所有元素都被置零。对于 ndim 超过 2 的数组，tril 将应用于最后两个轴。

参数:
m类数组, 形状 (…, M, N)

输入数组。

kint, 可选

要置零的对角线以上的部分。k = 0 (默认值) 表示主对角线，k < 0 表示主对角线以下，k > 0 表示主对角线以上。

返回:
trilndarray, 形状 (…, M, N)

m 的下三角部分，与 m 具有相同的形状和数据类型。

另请参阅

triu

类似的功能，但用于上三角部分

示例

>>> import numpy as np
>>> np.tril([[1,2,3],[4,5,6],[7,8,9],[10,11,12]], -1)
array([[ 0,  0,  0],
       [ 4,  0,  0],
       [ 7,  8,  0],
       [10, 11, 12]])

>>> np.tril(np.arange(3*4*5).reshape(3, 4, 5))
array([[[ 0,  0,  0,  0,  0],
        [ 5,  6,  0,  0,  0],
        [10, 11, 12,  0,  0],
        [15, 16, 17, 18,  0]],
       [[20,  0,  0,  0,  0],
        [25, 26,  0,  0,  0],
        [30, 31, 32,  0,  0],
        [35, 36, 37, 38,  0]],
       [[40,  0,  0,  0,  0],
        [45, 46,  0,  0,  0],
        [50, 51, 52,  0,  0],
        [55, 56, 57, 58,  0]]])
本页内容