python - 2次元マトリックスを3次元ワンホットマトリックスnumpyに変換する

Question

np 行列があり、要素を 3 次元として 1 つのホット エンコーディングで 3D 配列に変換したいと考えています。各行をループせずに行う方法はありますか?

a=[[1,3],
   [2,4]]

にする必要があります

b=[[1,0,0,0], [0,0,1,0],
   [0,1,0,0], [0,0,0,1]]

Answer 1

np.eye (恒等行列) と numpy の強力なインデックス作成を使用した最もシンプルでエレガントなソリューションを次に示します。

labels_3d = np.eye(N_CLASSES)[labels_2d]