利用np.eye(n)(x)进行one-hot化

np.eye(3)生产一个单位矩阵3*3的单位矩阵

当np.eye(3)后面跟一个整数i时,此时取得是该矩阵的第i行。

tp1=np.eye(3)
tp1=[[1,0,0],
    [0,1,0],
    [0,0,1]]

tp1[2]取该矩阵的第2行,为 [0,0,1]

当np.eye(3)后面跟一个数列时,此时取得是该矩阵的多个行

tp1[1][2]取的是矩阵索引为1,2的行;分别为  

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

该结果恰好是对索引的one-hot处理

 

tp1=([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

tp1[[1,2]]表示的是取tp1的第1行和第2行


tp[[1,2],[0,1]]表示的取tp1的[1][0]位置的元素和tp1的[2][1]位置的元素

等同于tp1[1][0]  tp1[2][1]

 

你可能感兴趣的:(python)