NumPy 中的 One-Hot 编码索引数组
在 NumPy 中,将一维索引数组转换为 one-hot 编码的二维数组是一项共同的任务。例如,给定索引为 [1, 0, 3] 的数组 a,我们希望将其编码为:
b = [[0,1,0,0], [1,0,0,0], [0,0,0,1]]
要实现此目的,有两个关键步骤:
这是一个代码示例来说明:
<code class="python">import numpy as np a = np.array([1, 0, 3]) b = np.zeros((a.size, a.max() + 1)) b[np.arange(a.size), a] = 1 print(b)</code>
输出:
[[0. 1. 0. 0.] [1. 0. 0. 0.] [0. 0. 0. 1.]]
此方法有效地将索引数组转换为 one-hot 编码数组,其中每一行代表 a 中相应索引的 one-hot 编码值。
以上是如何在 NumPy 中将索引数组转换为 One-Hot 编码数组?的详细内容。更多信息请关注PHP中文网其他相关文章!