csv文件操作
1.什么是csv文件
csv文件叫做逗号分隔值文件 — 每一行内容是通过逗号来区分出不同列的
csv文件可以直接通过excel打开,以行列的形式保存和显示数据文件。但是相对于excel文件,它只能存储数据,不能保存公式和函数。
2.csv读操作
import csv
1.创建打开csv文件
文件对象 = open()
操作文件
文件对象.close()
with open() as 文件对象
操作文件
f = open('电影.csv', 'r', encoding='utf-8')
2.创建reader获取文件内容
- csv.reader(文件对象) — 获取文件内容,并且以列表为单位返回每一行内容
- csv.DictReader(文件对象) — 获取文件内容,并且以字典为单位返回第二行开始的每一行内容(字典的键是第一行的内容)
reader1 = csv.reader(f)
print(list(reader1))
reader2 = csv.DictReader(f)
print(list(reader2))
f.close()
练习:计算各城市的平均薪资
from re import findall
f1 = open('lagou.csv', 'r', encoding='utf-8')
reader1 = csv.DictReader(f1)
citys = {
}
for job in reader1:
c = job['city']
m = job['salary']
result = findall(r'(\d+)k-(\d+)k', m)
if not result:
continue
start, end = result[0]
moneys = citys.get(c, [])
moneys.append((int(start) + int(end))/2)
citys[c] = moneys
for key in citys:
print(key