一行代码对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中的对应操作方法,最终顺利解决,如有更优解请指教。

你可能感兴趣的:(一行代码对0-9进行独热编码转换)