将索引数组转换为1-hot编码的numpy数组

作者: 狗头军师
发布时间: 2024-08-19 05:54:00 (8天前)

2 条回复

1#
回复此人
v-star*위위 | 2020-08-24 15-08

您的数组a定义了输出数组中非零元素的列。您还需要定义行，然后使用花式索引： ``` >>> a = np.array([1, 0, 3]) >>> b = np.zeros((a.size, a.max()+1)) >>> b[np.arange(a.size),a] = 1 >>> b array([[ 0., 1., 0., 0.], [ 1., 0., 0., 0.], [ 0., 0., 0., 1.]]) ```

编辑

登录后才能参与评论