【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]]

这样就搞定了！

参考链接：
http://www.30daydo.com/article/376