Python按指定列读取csv文件及写文件时的隔行问题

编程环境：Python3.5

先讲一下如何读写csv文件：

import csvcsvfile=open('train.csv','r')  #Python3.5这里不要用rbreader=csv.reader(csvfile)train=[]for line in reader:    train.append(line)csvfile.close()csvfile = open('csv_test.csv','w') #同样Python3.5这里不要用wbwriter = csv.writer(csvfile)for line in train:   writer.writerow(line)csvfile.close()

这样便可以读写CSV文件了，但写入CSV文件时会出现隔行问题，如下图：

具体原因请参考：  python3 writerow CSV文件多一个空行

解决方法：

将csvfile=open('csv_test.csv','w')这行代码改为csvfile=open('csv_test.csv','w',newline='')即可。

接下来说明如何读写指定行列的数据。

举个例子：

这是mnist数据集的部分数据，我的目的是读取第二行第二列之后的数据存入列表中，并读取第一列的所有数据存入另一个列表。即先去除第一行标签，再按列存储到不同列表。

具体代码如下：

import csvcsvfile=open('train.csv','r')reader=csv.reader(csvfile)i=0train=[]label=[]for line in reader:   if i<1:       i+=1   #这里的作用是把第一行去掉   else:       train.append(line[1:])   #读取第二列以后的数据       label.append(line[0])   #读取第一列的数据csvfile.close()#以下为把train,label的内容写进csv文件，验证读取是否正确csvfile = open('csv_test.csv','w',newline='')writer = csv.writer(csvfile)for line in train:   writer.writerow(line)csvfile.close()csvfile = open('csv_testt.csv', 'w',newline='')writer = csv.writer(csvfile)for line in label:   writer.writerow(line)csvfile.close()

结果展示：

csv_test.csv的数据如下，即把第一行第一列剔除后的结果：

csv_testt.csv的数据如下，即把第一行剔除后选取第一列的结果：

这里只是举个例子，当然可以选取其他行列，看自己需求修改相应代码即可。