python中eye函數(shù)的作用是什么

小樊
84
2024-09-08 21:49:07

在Python中，eye函數(shù)通常指的是NumPy庫(kù)中的numpy.eye函數(shù)

numpy.eye函數(shù)用于創(chuàng)建一個(gè)對(duì)角線上元素為1，其他元素為0的二維數(shù)組（也稱為單位矩陣）。函數(shù)的基本語(yǔ)法如下：

numpy.eye(N, M=None, k=0, dtype=float, order='C', *, like=None)

參數(shù)說明：

  • N：輸出矩陣的行數(shù)。
  • M：輸出矩陣的列數(shù)。默認(rèn)值為None，表示列數(shù)等于行數(shù)。
  • k：指定對(duì)角線的索引。正值表示上對(duì)角線，負(fù)值表示下對(duì)角線，0表示主對(duì)角線。
  • dtype：輸出矩陣的數(shù)據(jù)類型。默認(rèn)為float。
  • order：輸出矩陣的存儲(chǔ)順序。'C'表示按行存儲(chǔ)，'F'表示按列存儲(chǔ)。
  • like：用于指定輸出矩陣的類型。如果提供了like參數(shù)，則輸出矩陣將與like參數(shù)指定的矩陣具有相同的類型。

以下是一個(gè)使用numpy.eye函數(shù)的示例：

import numpy as np

# 創(chuàng)建一個(gè)3x3的單位矩陣
identity_matrix = np.eye(3)
print(identity_matrix)

輸出結(jié)果：

[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]

這個(gè)示例中，我們創(chuàng)建了一個(gè)3x3的單位矩陣，其對(duì)角線上的元素為1，其他元素為0。

0 踩