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

原创于 2020-06-24 16:58:45 发布 · 1.3k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#python #csv

我的疑问与解答专栏收录该内容

6 篇文章

订阅专栏

本文介绍如何使用NumPy对MNIST数据集进行加载与重塑，将其转换为适用于卷积神经网络输入的28×28×1矩阵格式，涉及loadtxt()与savetxt()函数的参数详解。

示例

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表示不知道会产生多少个这样的矩阵，个数不指定

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