作用1：生成對角陣，主對角線上元素為1，其余位置均為0

import numpy as np a = np.eye(3) print(a)

結果：

如果要讀取第二行，則寫作：np.eye(3)[1] ，結果為：[0,1,0]

作用2：形成one-hot編碼

import numpy as np x = np.array([1,2,0,1,2]) a = np.eye(3)[x] print(a)

結果：

?【后記】：python一行代碼可以做很多事，沒有經過系統學習，遇到很多問題，看都看不懂，還是要遇到一點問題就解決一點問題，漸漸的經驗積累多了，經常用著，就熟悉了。

總結

以上是生活随笔為你收集整理的python | np.eye()函数的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。