Day 29：【99天精通Python】数据持久化 - CSV与Excel

Day 29：【99天精通Python】数据持久化 - CSV与Excel - 办公自动化的第一步

前言

欢迎来到第29天！

在昨天的爬虫练习中，我们成功从网页上抓取了数据。但这些数据如果只打印在终端里，关掉窗口就没了，实在太可惜。

在职场中，最通用的数据交换格式莫过于CSV和Excel。老板和同事通常只认 Excel 表格。如果你能把程序跑出来的结果直接生成一个漂亮的 Excel 报表，那绝对是职场加分项。

今天，我们将学习如何用 Python 读写这两种格式的文件。这不仅是爬虫的后续，更是办公自动化 (RPA)的基础。

本节内容：

CSV 文件读写 (内置csv模块)
DictReader与DictWriter的妙用
Excel 文件读写 (第三方库openpyxl)
操作工作簿 (Workbook)、工作表 (Sheet)、单元格 (Cell)
实战练习：将爬取的电影数据存入 Excel

一、CSV 文件操作：简单高效

1.1 什么是 CSV？

CSV (Comma-Separated Values) 即逗号分隔值文件。它本质上就是纯文本，每行一条记录，字段之间用逗号分开。
优点：通用、轻量、可以用 Excel 打开。

1.2 写入 CSV

Python 内置了csv模块。

importcsv data=[["姓名","年龄","城市"],["Alice",25,"New York"],["Bob",30,"London"],["Charlie",22,"Beijing"]]# newline='' 非常重要，防止Windows下出现空行withopen("people.csv","w",encoding="utf-8-sig",newline='')asf:writer=csv.writer(f)# 写入单行 (表头)writer.writerow(data[0])# 写入多行 (数据)writer.writerows(data[1:])print("CSV 写入完成！")

小贴士：编码建议使用utf-8-sig，这样 Excel 打开时才不会乱码（BOM头）。

1.3 读取 CSV

importcsvwithopen("people.csv","r",encoding="utf-8-sig")asf:reader=csv.reader(f)forrowinreader:print(row)# 输出: ['姓名', '年龄', '城市'] ...

1.4 像字典一样操作 (推荐)

使用DictWriter和DictReader，代码可读性更高。

# --- 字典写入 ---headers=["id","name","score"]rows=[{"id":101,"name":"Tom","score":88},{"id":102,"name":"Jerry","score":95}]withopen("scores.csv","w",encoding="utf-8-sig",newline='')asf:writer=csv.DictWriter(f,fieldnames=headers)writer.writeheader()# 写入表头writer.writerows(rows)# --- 字典读取 ---withopen("scores.csv","r",encoding="utf-8-sig")asf:reader=csv.DictReader(f)forrowinreader:print(f"{row['name']}的分数是{row['score']}")

二、Excel 操作：OpenPyXL

虽然 CSV 很好用，但它不支持格式（颜色、字体、合并单元格）。如果需要操作真正的.xlsx文件，我们需要openpyxl库。

安装：

pipinstallopenpyxl

2.1 创建并写入 Excel

fromopenpyxlimportWorkbook# 1. 创建工作簿wb=Workbook()# 2. 获取激活的工作表 (默认第一个)ws=wb.active ws.title="我的报表"# 3. 写入表头ws['A1']="商品"ws['B1']="价格"ws['C1']="销量"# 4. 追加数据 (append 会自动寻找下一行)ws.append(["iPhone 15",5999,100])ws.append(["小米 14",3999,200])ws.append(["华为 Mate60",6999,150])# 5. 保存文件wb.save("sales_report.xlsx")print("Excel 保存成功！")

2.2 读取 Excel

fromopenpyxlimportload_workbook# 1. 加载现有的工作簿wb=load_workbook("sales_report.xlsx")# 2. 获取指定的工作表ws=wb["我的报表"]# 或者 ws = wb.active# 3. 读取单元格print(ws['A2'].value)# iPhone 15# 4. 遍历所有行print("\n--- 遍历数据 ---")# values_only=True 直接返回数据，而不是 Cell 对象forrowinws.iter_rows(min_row=2,values_only=True):product,price,sales=rowprint(f"商品:{product}, 销售额:{price*sales}")

2.3 简单的样式设置

fromopenpyxl.stylesimportFont,PatternFill# 加载并修改wb=load_workbook("sales_report.xlsx")ws=wb.active# 设置 A1 单元格：加粗、红色字体ws['A1'].font=Font(bold=True,color="FF0000")# 设置 B1 单元格：黄色背景ws['B1'].fill=PatternFill("solid",fgColor="FFFF00")wb.save("sales_report_styled.xlsx")

三、Pandas：终极武器 (预告)

我们在后面的【实战篇】会专门讲 Pandas。这里先提一句：Pandas 读写 Excel 只需要一行代码。

# 伪代码预览importpandasaspd# 读df=pd.read_excel("data.xlsx")# 写df.to_excel("output.xlsx",index=False)

但openpyxl依然很重要，因为 Pandas 不擅长精细的样式调整，而openpyxl可以。

四、实战练习：豆瓣电影入库

结合Day 28 (BeautifulSoup)的代码，我们将爬取的豆瓣电影数据保存为 Excel 文件。

importrequestsfrombs4importBeautifulSoupfromopenpyxlimportWorkbookdefsave_douban_to_excel():# 1. 准备 Excelwb=Workbook()ws=wb.active ws.title="豆瓣Top250"ws.append(["排名","电影名","评分","引言"])# 表头# 2. 爬取数据 (简单演示第一页)url="https://movie.douban.com/top250"headers={"User-Agent":"Mozilla/5.0 ... (省略)"}# 记得补全 UAtry:# 为了演示，这里假设请求成功# 实际代码请补全 requests 逻辑# ...# 模拟一些数据 (假设这是爬下来的)movies=[(1,"肖申克的救赎",9.7,"希望让人自由。"),(2,"霸王别姬",9.6,"风华绝代。"),(3,"阿甘正传",9.5,"人生就像一盒巧克力。")]forrank,title,rating,quoteinmovies:ws.append([rank,title,rating,quote])# 3. 保存filename="douban_top250.xlsx"wb.save(filename)print(f"数据已保存至{filename}")exceptExceptionase:print(f"Error:{e}")save_douban_to_excel()

五、常见问题

Q1：`PermissionError: [Errno 13] Permission denied`？

这是因为你打开了 Excel 文件。Windows 锁定了文件，Python 无法写入。
解决：关掉 Excel 软件，再运行代码。

Q2：CSV 中文乱码？

假如用 Excel 打开乱码：保存时使用encoding="utf-8-sig"。
假如是 Pandas 读取乱码：尝试encoding="gbk"。

Q3：`openpyxl`能处理`.xls`(老版Excel) 吗？

不能。openpyxl只支持.xlsx。如果要处理.xls，需要用xlrd(读) 和xlwt(写) 库。但建议统统转为.xlsx。

六、小结

关键要点：

CSV是程序员的最爱，简单、通用。
Excel是老板的最爱，带样式、易阅读。
写入 CSV 记得newline=''和utf-8-sig。
写入 Excel 记得最后wb.save()。

七、课后作业

成绩统计表：创建一个 CSV 文件scores.csv，包含 “姓名,语文,数学,英语”。读取该文件，计算每个人的总分，并将结果（姓名,总分）写入一个新的 Excel 文件result.xlsx。
九九乘法表：使用openpyxl生成一个 Excel 文件，在 Sheet 中打印出九九乘法表。
日志转换：读取 Day 12 中生成的diary.txt，将其内容解析后转存为 Excel，两列：[时间, 内容]。