numpy.tril

numpy.tril(m, k=0)

陣列的下三角部分。

傳回一個陣列的副本，其中高於第 k 條對角線的元素皆設為零。對於 ndim 超過 2 的陣列，tril 將應用於最後兩個軸。

參數:

marray_like，形狀 (…, 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]]])