Python与Pandas和XlsxWriter组合工作 – 2

继续回答“详细讲解Python与Pandas和XlsxWriter组合工作”的第二部分。

在使用Pandas和XlsxWriter生成Excel文件之前，我们需要先安装它们。在命令行中运行如下指令即可：

pip install pandas
pip install xlsxwriter

接下来，我们需要创建一个Pandas数据帧，并将其写入Excel文件中。我们可以通过Pandas提供的DataFrame类来创建数据帧。如下所示，我们创建一个包含学生姓名、年龄和语文、数学、英语成绩的数据帧。

import pandas as pd

df = pd.DataFrame({
  'Name': ['Tom', 'Jerry', 'Mike', 'John', 'Mary'],
  'Age': [18, 19, 18, 17, 19],
  'Chinese Score': [80, 85, 92, 77, 88],
  'Math Score': [95, 92, 87, 90, 78],
  'English Score': [81, 88, 90, 83, 92]
})

创建好了数据帧后，我们可以使用XlsxWriter来把它写入Excel文件中。如下所示，我们使用XlsxWriter的Workbook类来创建一个Excel文件。然后使用数据帧的to_excel()方法把数据写入到Excel文件中。

import xlsxwriter

# 创建一个Excel文件对象
workbook = xlsxwriter.Workbook('students.xlsx')

# 创建一个工作表对象
worksheet = workbook.add_worksheet()

# 标题格式
header_format = workbook.add_format({'bold': True, 'border': 1})

# 写入列标题
worksheet.write(0, 0, 'Name', header_format)
worksheet.write(0, 1, 'Age', header_format)
worksheet.write(0, 2, 'Chinese Score', header_format)
worksheet.write(0, 3, 'Math Score', header_format)
worksheet.write(0, 4, 'English Score', header_format)

# 写入数据
df.to_excel(workbook, sheet_name='Students', startrow=1, startcol=0, index=False)

# 关闭文件
workbook.close()

在上面的代码中，我们首先创建了一个Excel文件对象，然后使用add_worksheet()方法创建了一个工作表对象。接着，我们定义了一个header_format变量，它是用来设置列标题格式的。然后，我们使用工作表对象的write()方法把列标题写入到工作表中。

接着，我们使用数据帧的to_excel()方法把数据写入Excel文件中。其中，sheet_name参数用于指定工作表的名称，startrow和startcol参数分别用于指定写入数据的起始行和起始列，index参数用于指定是否包含数据帧的索引（默认为True）。

最后，我们使用工作表对象的close()方法关闭Excel文件。

通过上述代码，我们就可以创建一个包含学生信息的Excel文件了。其中，数据帧中的数据将被自动写入工作表中。