1、csv文件的写入
import csv
'''写入csv文件'''
# encoding:编码格式, newline=""可让数据不用隔行写入
with open('demo.csv','w',encoding='utf-8',newline="",) as datacsv:
csv_writer = csv.writer(datacsv, dialect=("excel"),)
csv_writer.writerow(['A','B','C','D','E','F'])
csv_writer.writerow(['1','2','3','4','5','6'])
结果:
2、csv文件的读
'''读csv文件'''
f = open('demo.csv', 'r')
reader = csv.reader(f)
# print(next(reader)) # 读取一行
# print(next(reader)) # 读取下一行
# for循环读取
for i in reader:
print(i)
结果:
3、panda库的读写文件(更简单)
'''pandas库读写操作'''
import pandas as pd
filename = 'demo.csv'
# 读文件
df = pd.read_csv(filename)
print(df)
# 取出df的前5行
data = df.head()
# 写入文件
data.to_csv('demo_01.csv')
结果:
发现pandas的写入操作多了一点东西:第一行中的逗号和第二行中的0(如果下面还有数据的话,肯定还有1、2、3…),这些叫做index索引,如果想去除csv文件中的index索引,只需在写入操作时添加一句 index=False 即可。
data.to_csv('demo_01.csv',index=False)
结果: