Pythonのnumpyのワンホット

Question

ワンホット配列を解読する方法を、

つまり

array([[0, 0, 1], 
     [2, 1, 0]]) 

として

array([[[1, 0, 0], 
     [1, 0, 0], 
     [0, 1, 0]], 

     [[0, 0, 1], 
     [0, 1, 0], 
     [1, 0, 0]]]) 

このワンホットエンコードされた行列を作るための最善の方法は何ですか？サンプル実行

a.argmax(2) 

- -

Answer 1

使用np.argmaxaxis=2に沿っ

In [186]: a 
Out[186]: 
array([[[1, 0, 0], 
     [1, 0, 0], 
     [0, 1, 0]], 

     [[0, 0, 1], 
     [0, 1, 0], 
     [1, 0, 0]]]) 

In [187]: a.argmax(2) 
Out[187]: 
array([[0, 0, 1], 
     [2, 1, 0]])