【python】读写csv文件、从csv文件加载数据集

在E盘创建文件夹csv， csv中创建csv文件csvFile：

 

读取csv文件：

import csv

csvfile = open('E:\csv\csvFile.csv', 'r')
reader = csv.reader(csvfile)
a = list(reader)
print(a)

 读出结果显示在一行， 有些拥挤， 所以可以一行一行读：

with open('E:\csv\csvFile.csv', 'r') as f:
    reader = csv.reader(f)
    for row in reader:
        print(row)

 

 

写csv文件：

with open('E:\csv\csvFile.csv', 'w', newline="") as f:
    writer = csv.writer(f)
    writer.writerow(['计组', '110', '93'])
    writer.writerow(['离散', '50', '98'])

 

 

 

 发现写入行把之前的覆盖掉了， 怎么办呢？（当然对于新建的文件，这样添加行还是很推荐的） 可以把open的第二个参数'w'改为'a'。

with open('E:\csv\csvFile.csv', 'a', newline="") as f:
    writer = csv.writer(f)
    writer.writerow(['计组', '110', '93'])
    writer.writerow(['离散', '50', '98'])

 

 

 

 还有一点需要注意的是open的第三个参数newline，这里设置newline=""，如果不设置newline=""的话每个写入行之间会出现空行：

 

从csv文件加载数据集：

import csv
import pandas as pd

csvfile = open('E:\csv\csvFile.csv', 'r')
data = pd.read_csv(csvfile)
print(data)