X = np.expand_dims(df.values[:, 0:18].astype(float), axis=2)
时间: 2024-05-20 15:18:36 浏览: 40
基于numpy中的expand_dims函数用法
这段代码使用了NumPy库中的expand_dims函数,将一个形状为(N,18)的二维数组转化为形状为(N,18,1)的三维数组。具体来说,这个数组是通过将原数组的第二个维度(即列数)扩展了一个维度得到的。astype(float)将原数组中的数据类型转换为浮点型。这个操作通常用于将数据输入到深度学习模型中,因为深度学习模型需要输入三维数组作为输入数据。
阅读全文