Python numpy matlib.identity用法及代碼示例

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

用法:

matlib.identity(n, dtype=None)

返回給定大小的平方單位矩陣。

參數：

n： int

返回的單位矩陣的大小。

dtype： 數據類型，可選

輸出的數據類型。默認為 float 。

返回：

out： 矩陣

n x n 矩陣，其主對角線設置為 1，所有其他元素為零。

例子：

>>> import numpy.matlib
>>> np.matlib.identity(3, dtype=int)
matrix([[1, 0, 0],
        [0, 1, 0],
        [0, 0, 1]])