Python读CSV数据

Python读CSV数据

Python读取CSV数据有两种：

1.按行读取csv(使用CSV库）

2.按列读取csv(使用Pandas库）

下面介绍第一种，逐行读取

1.按行读取csv(使用CSV库）

代码如下：

import csv if __name__ =='__main__': filePath="test.csv" with open(filePath, 'r',encoding="utf-8") as csvfile: reader = csv.reader(csvfile) for row in reader: #the first row is table header print(row) #type:list print(type(row))

第一个打印的row为表头的内容，之后会逐行打印所有的内容，并且row的类型为list。

2.按列读取csv(使用Pandas库）

这里以test.csv文件为例，文件一共包含两列（含表头），两列的名字分别为parameter和importance。接下来用pandas库按列读取，并分别存入到list中。

 代码如下：

import pandas as pd if __name__ =='__main__': filePath="test.csv" df=pd.read_csv("test.csv") parameterList=df["parameter"].values.tolist() importanceList=df["importance"].values.tolist() print(parameterList) print(importanceList)