numpy.tri#

numpy.tri(N, M=None, k=0, dtype=<class 'float'>, *, like=None)[原始碼]#

一個在給定對角線及其下方為 1，其他地方為 0 的陣列。

參數:

Nint: 陣列中的列數。
Mint, optional: 陣列中的行數。預設情況下，M 等於 N。
kint, optional: 子對角線，在其及其下方陣列會被填滿。 k = 0 是主對角線，而 k < 0 是在其下方，且 k > 0 是在其上方。預設值為 0。
dtypedtype, optional: 傳回陣列的資料型別。預設值為 float。
likearray_like, optional: 參考物件，允許建立非 NumPy 陣列的陣列。如果作為 like 傳入的類陣列物件支援 __array_function__ 協定，則結果將由其定義。在這種情況下，它確保建立一個與透過此引數傳入的物件相容的陣列物件。

版本 1.20.0 新增。

傳回值:

trindarray，形狀為 (N, M): 陣列，其下三角形填滿 1，其他地方為零；換句話說，對於 j <= i + k，T[i,j] == 1，否則為 0。

範例

>>> import numpy as np
>>> np.tri(3, 5, 2, dtype=int)
array([[1, 1, 1, 0, 0],
       [1, 1, 1, 1, 0],
       [1, 1, 1, 1, 1]])

>>> np.tri(3, 5, -1)
array([[0.,  0.,  0.,  0.,  0.],
       [1.,  0.,  0.,  0.,  0.],
       [1.,  1.,  0.,  0.,  0.]])