? ? 首先生成一個主對角全為1的其余全為0的矩陣，比如有n個分類就是n * n，效果如下：

? ? 隨后根據標簽列表（或者numpy數組）選取合適的行，比如標簽是[9, 1, 0, 0]， 那么就會選擇上圖矩陣中對應的9、1、0、0行，得到one_hot標簽，如果不熟悉numpy數組的列表切片的（就是說numpy_array[slice]中的slice是列表）?，可以看下這篇Python Numpy數組使用列表索引

? ? 恢復的話就是找列表中為1的下標即可。

? ? 代碼如下：

            
              # encoding = utf-8
'''
    author : James-J
    time : 2019/05/29
'''

import numpy as np

if __name__ == '__main__':
    one_hot = np.eye(10) # 10*10的矩陣 對角線上是1
    print('np.eye(10)\n', one_hot)
    # 兩種方法 傳一維的numpy數組和列表都可以
    label = np.array([1, 4, 8, 9, 5, 0])
    one_hot_label = one_hot[label.astype(np.int32)] # 表示選取矩陣上面的第幾行
    # label = [1, 4, 8, 9, 5, 0]
    # one_hot_label = one_hot[label]
    print('-----------------one_hot--------------------')
    print(one_hot_label)

    label = [one_label.tolist().index(1) for one_label in one_hot_label] # 找到下標是1的位置
    print('------------------label---------------------')
    print(label)
            
          

? ? 得到的結果：

?

?