Python numpy tri用法及代碼示例

本文簡要介紹 python 語言中 numpy.tri 的用法。

用法:

numpy.tri(N, M=None, k=0, dtype=<class 'float'>, *, like=None)

一個數組，在給定對角線處和下方都有一個，在其他地方有零。

參數：

N： int

數組中的行數。

M： 整數，可選

數組中的列數。默認情況下，M 等於 N。

k： 整數，可選

sub-diagonal 填充數組的位置和下方。 k = 0 是主對角線，而 k < 0 在其下方， k > 0 在其上方。默認值為 0。

dtype： dtype，可選

返回數組的數據類型。默認為浮點數。

like： array_like

允許創建非 NumPy 數組的引用對象。如果作為 like 傳入的類似數組支持 __array_function__ 協議，則結果將由它定義。在這種情況下，它確保創建一個與通過此參數傳入的數組對象兼容的數組對象。

返回：

tri： 形狀的ndarray(N，M)

其下三角形在其他地方填充 1 和 0 的數組；換句話說 T[i,j] == 1 為 j <= i + k ，否則為 0。

例子：

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