Python记录numpy.empty()函数引发的问题及解决

问题

最近用深度学习训练一个分类模型，在划分训练集和测试集时遇到一个问题。

意思是list的索引必须是整形，不能是float型。

经排查发现

存储标签的list是使用np.empty()函数创建的

这个函数可以创建指定shape的list，但是创建的并非是一个空的多维数组

如 np.empty((3,3))，默认为numpy.float64型，而实际标签值为整形

实际返回结果为：

print(np.empty((3,3)))

[[6.95177882e-310 6.95177882e-310 0.00000000e+000]
 [0.00000000e+000 6.95177883e-310 0.00000000e+000]
 [0.00000000e+000 0.00000000e+000 1.21698002e-152]]

查阅资料，发现np.empty()返回一个随机元素的矩阵，大小按照参数定义。

解决方法

其实使用在使用np.empty()创建list，可以指定类型

如：

print(np.empty(3,3,dtype=int))

[[4128860 6029375 3801156]
 [4391004 6553711 7536741]
 [5242972 7602297 7274600]]

总结

这样就搞定了！

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。