使用np.eye生成M*N矩陣，后面跟的數組說明1偏移的位置

import numpy as np values = np.array([1, 0, 3]) n_values = np.max(values) + 1 print(np.eye(n_values)[values])# 輸出結果 array([[ 0., 1., 0., 0.],[ 1., 0., 0., 0.],[ 0., 0., 0., 1.]])

參考

https://blog.csdn.net/zhongranxu/article/details/79332154

總結

以上是生活随笔為你收集整理的使用numpy生成one_hot的全部內容，希望文章能夠幫你解決所遇到的問題。

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