python | np.eye()函数

作用1：生成对角阵，主对角线上元素为1，其余位置均为0

import numpy as np
a = np.eye(3)
print(a)

结果：

如果要读取第二行，则写作：np.eye(3)[1] ，结果为：[0,1,0]

作用2：形成one-hot编码

import numpy as np
x = np.array([1,2,0,1,2])
a = np.eye(3)[x]
print(a)

结果：

 【后记】：python一行代码可以做很多事，没有经过系统学习，遇到很多问题，看都看不懂，还是要遇到一点问题就解决一点问题，渐渐的经验积累多了，经常用着，就熟悉了。