一些简单的csv文件读写操作

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)

结果:
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_43750528/article/details/130408518