python对csv数据文件进行读取、重构、保存

示例

import numpy as np 

a = np.loadtxt('mnist_test.csv', usecols=np.arange(2,786), delimiter=',', skiprows=1)
a.reshape(-1,28,28,1)
b = np.savetxt('mnist_test_reshape.csv', a, fmt='%d', delimiter=',')

说明

loadtxt()：读取文件
savetxt()：保存文件
usecols：读取的列，np.arange(2,786)表示从第三列读到第786列，也可单独指定某些列
delimiter：分割符
skiprows：跳过的行，skiprows=1跳过第一行，从第二行开始读取
fmt：保存格式
reshape(-1,28,28,1):将a转化为28×28×1的矩阵，-1表示不知道会产生多少个这样的矩阵，个数不指定

这样的数据就既可作为卷积神经网络的输入