Pandas Python: 将图片和数据框写入同一Excel文件

在本文中，我们将介绍如何将Pandas中的数据框和图像一起写入同一Excel文件中。我们将演示如何在Excel电子表格中创建工作表，并将数据框和图像写入其中。我们还将介绍如何在工作表中添加数据框和图像，如何对这些数据进行格式化和修改。

准备工作

在开始之前，请确保已安装以下库：

Pandas
openpyxl
Pillow

您可以使用以下命令在终端中安装这些库：

!pip install pandas openpyxl Pillow

请确保您在运行此代码之前已将所有必需的图像和数据框准备好。我们将使用以下示例数据。

示例数据

我们将使用以下示例数据：

两幅图像，分别命名为“image1.png”和“image2.png”
两个数据框，分别命名为“df1”和“df2”

请确保所有文件都位于同一文件夹中。

创建Excel表格

首先，让我们创建一个Excel电子表格。我们将使用Pandas库创建一个空的工作簿，并将其保存为Excel文件。

import pandas as pd

# 创建一个空的Excel电子表格
writer = pd.ExcelWriter('output.xlsx', engine='openpyxl')

# 将电子表格保存为Excel文件
writer.save()

该电子表格将保存在当前工作目录中。

写入数据框

现在，我们将向电子表格中添加一个名为“DataFrame1”的工作表，并将数据框“df1”写入该工作表。

# 将数据框写入电子表格
df1.to_excel(writer, sheet_name='DataFrame1', index=False)

# 保存电子表格
writer.save()

我们使用to_excel()函数将数据框写入Excel表格。请注意，我们将index参数设置为False，因为我们不想将行索引写入电子表格。

现在，我们可以打开保存的电子表格，看看数据框是否已成功写入。

写入图像

现在，让我们向电子表格添加一个名为“Image1”的工作表，并将图像“image1.png”写入该工作表。

from openpyxl.drawing.image import Image

# 加载图像
img = Image('image1.png')

# 在电子表格上添加一个工作表以保存图像
wb = writer.book
wb.create_sheet('Image1')

# 将图像写入Excel表格
ws = wb['Image1']
ws.add_image(img, 'A1')

# 保存电子表格
writer.save()

我们使用Openpyxl库的Image()函数加载图像。然后，我们使用writer.book对象创建一个名为“Image1”的新工作表，并在其中添加图像。

现在，我们可以打开保存的电子表格，看看图像是否已成功写入。

格式化数据

我们可以使用Pandas的ExcelWriter()方法来格式化数据框。以下示例演示如何使数据框中的所有单元格都居中对齐：

# 将数据框居中显示
df1.to_excel(writer, sheet_name='DataFrame1', index=False)
workbook = writer.book
worksheet = writer.sheets['DataFrame1']
center_alignment = workbook.add_format({'align': 'center'})
for col in df1:
    column = df1[col]
    max_len = len(max((str(s) for s in column.values), key=len))
    worksheet.set_column(df1.columns.get_loc(col), df1.columns.get_loc(col), max_len, center_alignment)
    worksheet.freeze_panes(1, 0)
writer.save()

我们首先将df1数据框写入名称为“DataFrame1”的工作表。然后，我们使用add_format()函数创建一个格式对象，并将其应用于数据框中的所有单元格。格式对象使所有单元格居中对齐。

请注意，set_column()方法用于设置指定列的宽度和格式。

修改数据

Pandas的ExcelWriter()方法还包括可用于修改现有电子表格的方法。下面是一个将工作表中的数据框替换为新数据框的示例：

# 将DataFrame1工作表中的数据框替换为新数据框
df_new = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df_new.to_excel(writer, sheet_name='DataFrame1', index=False)

# 保存电子表格
writer.save()

我们使用pd.DataFrame()函数创建一个新的数据框df_new。然后，我们将其写入名为“DataFrame1”的工作表。写入电子表格后，我们可以再次打开电子表格，看看数据框是否已成功修改。