numpy.tril #

麻木的。tril ( m , k = 0 ) [来源] #

数组的下三角形。

返回一个数组的副本,其中第 k个对角线上方的元素已为零。对于ndim超过 2 的数组,tril将应用于最后两个轴。

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

输入数组。

k整数,可选

其上方的对角线为零元素。 k = 0(默认值)为主对角线,k < 0位于其下方,k > 0位于其上方。

返回
tril ndarray,形状 (…, M, N)

m的下三角形,其形状和数据类型与m相同。

也可以看看

triu

同样的事情,仅适用于上三角形

例子

>>> 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]]])