Python读取pdf表格写入excel的方法

下面是Python读取pdf表格写入excel的方法的完整实例教程。

1. 环境准备

首先，我们需要安装三个Python库，分别是pdfplumber、openpyxl和os，可以通过pip命令安装：

!pip install pdfplumber
!pip install openpyxl

2. 实现步骤

接下来，我们具体来看如何使用Python实现读取pdf表格写入excel的功能。

2.1 读取pdf表格

我们可以使用pdfplumber库来读取pdf表格，它提供了比较方便的接口。

import pdfplumber

# 读取pdf文件
pdf = pdfplumber.open("data.pdf")

# 获取第一页的数据
first_page = pdf.pages[0]

# 获取第一页中的表格数据
table = first_page.extract_tables()[0]

# 打印表格数据
print(table)

在上面的代码中，我们首先使用pdfplumber的open方法打开pdf文件，并获取到第一页的数据。然后，我们使用extract_tables方法从第一页中提取表格数据，并选择第一个表格进行打印。

2.2 写入excel表格

接下来，我们将获取到的pdf表格数据写入到excel文件中，可以使用openpyxl库来实现。

import openpyxl

# 创建一个新的excel文件
workbook = openpyxl.Workbook()

# 获取第一个sheet
sheet = workbook.active

# 将表格数据写入excel中
for row in table:
    sheet.append(row)

# 保存excel文件
workbook.save("data.xlsx")

在上面的代码中，我们首先使用openpyxl的Workbook来创建一个新的excel文件，并获取到第一个sheet。然后，我们使用sheet的append方法将获取到的表格数据写入excel中。最后，我们使用save方法保存excel文件。

3. 完整实例

下面是一个完整的例子，将pdf中的第一个表格数据读取并写入excel文件。

import pdfplumber
import openpyxl

# 读取pdf文件
pdf = pdfplumber.open("data.pdf")

# 获取第一页的数据
first_page = pdf.pages[0]

# 获取第一页中的表格数据
table = first_page.extract_tables()[0]

# 创建一个新的excel文件
workbook = openpyxl.Workbook()

# 获取第一个sheet
sheet = workbook.active

# 将表格数据写入excel中
for row in table:
    sheet.append(row)

# 保存excel文件
workbook.save("data.xlsx")

4. 示例说明

示例1：读取pdf中的表格数据

假设我们有一个名为data.pdf的pdf文件，其中包含了一个表格数据，我们可以使用以下代码来读取表格数据：

import pdfplumber

# 读取pdf文件
pdf = pdfplumber.open("data.pdf")

# 获取第一页的数据
first_page = pdf.pages[0]

# 获取第一页中的表格数据
table = first_page.extract_tables()[0]

# 打印表格数据
print(table)

在上面的代码中，我们使用pdfplumber库来读取pdf文件，并获取到第一页中的表格数据。然后，我们使用print方法来打印表格数据。

示例2：将表格数据写入excel文件

使用示例1中获取到的表格数据，我们可以将其写入excel文件中，代码如下：

import openpyxl

# 创建一个新的excel文件
workbook = openpyxl.Workbook()

# 获取第一个sheet
sheet = workbook.active

# 将表格数据写入excel中
for row in table:
    sheet.append(row)

# 保存excel文件
workbook.save("data.xlsx")

在上面的代码中，我们使用openpyxl库来创建一个新的excel文件，并将表格数据写入excel中。最后，我们使用save方法来保存excel文件。