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

编程环境：Python3.5

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

import csv
csvfile=open('train.csv','r')  #Python3.5这里不要用rb
reader=csv.reader(csvfile)
train=[]
for line in reader:
    train.append(line)
csvfile.close()
csvfile = open('csv_test.csv','w') #同样Python3.5这里不要用wb
writer = 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 csv
csvfile=open('train.csv','r')
reader=csv.reader(csvfile)
i=0
train=[]
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的数据如下，即把第一行剔除后选取第一列的结果：

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