近日在Udacity學習Deep Learning,練習中遇到了需要轉換可能值為0-9的標簽至獨熱編碼的步驟。
這本來不是什么難事,但是群里的大神說可以用一行代碼實現,這就勾起了我的好奇心。。。
經過一個小時的資料搜集和python語法學習得到了以下版本:
return np.array( [ np.eye(10)[i] for i in x ] )
本人其實是個js黨,對python不是很熟悉,最后其實是轉換了思維讓我用js實現會怎么做,第一反應就是array.map,于是就去查了python中的對應操作方法,最終順利解決,如有更優解請指教。