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表示不知道會產生多少個這樣的矩陣，個數不指定

這樣的數據就既可作爲卷積神經網絡的輸入