1分钟看python如何处理CSV数据

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接： https://blog.csdn.net/LEEANG121/article/details/102604457

分析csv数据文件头

import csv
with open(('F:/LEEANG/深度学习/数据下载/eval.csv') as f:   #with函数用于自动关闭功能，open（）函数
#打开数据，实参包括数据的绝对路径
    reader = csv.reader(f)
    header_row = next(reader)   #函数next（）表示将返回文件的下一行。本代码中我们用了next（）一次，#因此得到的是文件的第一行
    print(header_row)

['survived', 'sex', 'age', 'n_siblings_spouses', 'parch', 'fare', 'class', 'deck', 'embark_town', 'alone']

打印某几列元素

# 继续之前代码

#添加新的代码
print (row[0:4])   #打印第一列到第五列所有元素
#注意，若将print的位置放置在header_row = next(reader)下面，则仅打印1到4列第二行元素

现在打印前三行内容

import csv   #导入csv模块
with open('E:/个人文件/学习/文件读取/train.csv', 'r') as f:  #从文件夹打开文件train.csv
    reader = csv.reader(f)   #读取文件并放入对象reader
    header = next(reader)    #读取第一行文件并返回给header，之后从下一行开始进行
    row = []    #创建一个空的行组数
    for line in reader:
        row.append(line)    #把行元素添加进row组
    print (row[0:3])    #打印第一至第三行（源文件的第一行已经去除）

[['0', 'male', '22.0', '1', '0', '7.25', 'Third', 'unknown', 'Southampton', 'n'], ['1', 'female', '38.0', '1', '0', '71.2833', 'First', 'C', 'Cherbourg', 'n'], ['1', 'female', '26.0', '0', '0', '7.925', 'Third', 'unknown', 'Southampton', 'y']]

打印前三行前三列

#接续上面的代码继续写
column = []    #建立空的列向量列表
for col in row:
    column.append(col[0:3])   #取前三列
#在新建的列向量列表中取前三行
for line in column:     
    row.append(line)
    row = row [0:3]
print(column)

[['0', 'male', '22.0'], ['1', 'female', '38.0'], ['1', 'female', '26.0']]