实现一个函数来对标称(类别)数据进行独热编码(One-Hot Encoding)。独热编码是一种将类别变量转换为机器学习算法可以更好地处理的数值型变量的方法。
输入描述:
函数`one_hot_encode`接收两个参数:1. x:一维numpy数组,包含需要编码的整数值2. n_col:可选参数,指定编码后的列数(类别数)   - 如果不提供,则自动设置为输入数组中的最大值+1


输出描述:
返回一个二维numpy数组:- 行数等于输入数组的长度- 列数等于n_col或自动确定的类别数- 每行只有一个1,其余为0元素类型行不需要修改,保持默认的即可。
示例1

输入

[0, 1, 2]

输出

[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]

备注:
1.对应的输入、输出已给出,您只用实现核心功能函数即可。2.支持numpy、scipy、pandas、scikit-learn库。
加载中...