【问题描述】
将pandas得到的数据保存到一个新工作表中,这个新工作表位于一个新的工作簿文件中。
【示例2-5】
本例使用的Excel文件的完整路径为“D:/Samples/ch02/04 数据输出到新的工作表/产品出厂价.xlsx”。该文件打开后如图2-4所示,是各种产品的品名、规格、单位和出厂价数据。本例导入文件的前6行数据,然后用pandas新建一个名为“产品出厂价2.xlsx”的Excel文件。该文件位于相同的路径下。将导入的数据添加到该文件的工作表Sheet1中。
图2-4 产品的出厂价
- 编写下面的代码:
code.python
import os
import pandas as pd
# 设置文件路径
file_path = "D:/Samples/ch02/04 数据输出到新的工作表/产品出厂价.xlsx"
# 导入Excel文件前6行数据,引擎为"openpyxl"
data = pd.read_excel(file_path, engine="openpyxl", nrows=6)
# 获取Excel文件所在目录的完整路径
dir_path = os.path.dirname(os.path.abspath(file_path))
# 新建ExcelWriter对象
writer = pd.ExcelWriter(os.path.join(dir_path, "产品出厂价2.xlsx"))
# 将导入的数据添加到该文件的工作表Sheet1中
data.to_excel(writer, sheet_name="Sheet1", index=False)
# 保存Excel文件
writer.save()
打开Python IDLE,新建一个脚本文件,将上面生成的代码复制进去,保存。运行脚本,新建一个名为“产品出厂价2.xlsx”的Excel文件。该文件位于相同的路径下。导入的6行数据添加到该文件的工作表Sheet1中。
【知识点扩展】
本例使用pandas的ExcelWriter函数新建Excel文件,用该函数的path参数指定文件保存路径和名称。