python之csv操作

在使用python爬虫时或者其他情况,都会用到csv存储与读取的相关操作,我们在这里就浅谈一下:

CSV(Comma-Separated Values)逗号分隔符,也就是每条记录中的值与值之间是用分号分隔的。

一、读文件

方法一:只是使用csv这个库

import csv # 导入csv这个库
from itertools import islice  # 当不读取csv文件第一行时,导入这个包有很棒的效果
# 读取china_city_aqi.csv文件,以r方式进行读取,编码是utf-8
with open("china_city_aqi.csv", "r", encoding = "utf-8") as f:
    # 将csv读取的文件放入reader中
    reader = csv.reader(f)
    # 之后进行遍历,islice(reader, 1, None) 中表示读取reader文件中的第一行至最后一行,当然你可以修改到第几行结束
    for line in islice(reader, 1, None):
        # 读取名称
        name = line[:1]
        # 读取名称之后的数据
        value = line[1:]
        print(name, ' ', value)

方法二:使用pandas这个库(非常方便了)

top_city.to_csv('top_city.csv', index=False, encoding='utf-8')
1、top_city:就是你要保存呢的数据,我的是列表保存
2、index:不想在文件中出现索引的话,那就设置为False,默认是True

二、写文件

import csv
headers = ['苹果', '香蕉', 'apple']
# newline:表示换行,默认情况下都是'\n'
file = open('writer.csv', 'w', encoding='utf-8', newline='')
writer = csv.writer(file)
writer.writerow(headers)