一些简单的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)

结果：