一行代碼對0-9進行獨熱編碼轉換

近日在Udacity學習Deep Learning,練習中遇到了需要轉換可能值為0-9的標簽至獨熱編碼的步驟。
這本來不是什么難事,但是群里的大神說可以用一行代碼實現,這就勾起了我的好奇心。。。
經過一個小時的資料搜集和python語法學習得到了以下版本:

return np.array( [ np.eye(10)[i] for i in x ] )

本人其實是個js黨,對python不是很熟悉,最后其實是轉換了思維讓我用js實現會怎么做,第一反應就是array.map,于是就去查了python中的對應操作方法,最終順利解決,如有更優解請指教。

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容