使用Python从PDF中提取饼图数据：基于图像处理的实战教程

新闻中心 NEWS CENTER

您当前位置：首页 > 新闻中心 > 网络学院

使用Python从PDF中提取饼图数据：基于图像处理的实战教程

2025-11-08

浏览次数：次

返回列表

使用python从pdf中提取饼图数据：基于图像处理的实战教程

本教程详细介绍了如何利用Python从PDF文档中提取饼图数据。核心方法是先将PDF页面转换为图像，然后借助OpenCV等图像处理库识别饼图的切片轮廓，并通过进一步的图像分析技术（如面积计算或颜色识别）来量化每个切片的数据。文章涵盖了库的安装、图像转换、轮廓检测及数据量化思路，旨在提供一个清晰、实用的数据提取解决方案。

在许多业务场景中，我们可能需要从PDF报告中自动化提取图表数据，例如饼图、柱状图等。由于PDF文档的复杂性，直接从其内部结构中解析图形数据通常非常困难。一种高效且灵活的策略是将PDF页面转换为图像，然后利用强大的图像处理库来识别和分析图表。本文将专注于如何使用Python实现这一过程，以饼图数据提取为例。

1. PDF到图像的转换

将PDF页面转换为图像是进行视觉分析的前提。Python生态系统提供了多个库来完成这项任务，其中pdf2image是一个流行且易于使用的选择。它依赖于Poppler工具集（在Linux上通常预装，在Windows和macOS上需要单独安装）。

安装pdf2image：

pip install pdf2image

基本转换示例：

pdf2image库的convert_from_path函数可以将PDF文件转换为PIL（Pillow）图像对象列表，每个图像对象代表PDF中的一页。

from pdf2image import convert_from_path
import os

def convert_pdf_to_images(pdf_path, output_folder="pdf_images"):
    """
    将PDF文件转换为一系列图像。
    :param pdf_path: PDF文件的路径。
    :param output_folder: 存储生成图像的文件夹。
    :return: 生成图像的文件路径列表。
    """
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    images = convert_from_path(pdf_path)
    image_paths = []
    for i, image in enumerate(images):
        image_path = os.path.join(output_folder, f"page_{i+1}.png")
        image.s*e(image_path, "PNG")
        image_paths.append(image_path)
    print(f"PDF '{pdf_path}' 已转换为 {len(images)} 张图像并保存到 '{output_folder}'。")
    return image_paths

# 示例用法
# 请将 'path/to/your/document.pdf' 替换为实际的PDF文件路径
# pdf_file = 'path/to/your/document.pdf' 
# image_files = convert_pdf_to_images(pdf_file)

注意事项： pdf2image在Windows环境下需要安装Poppler的二进制文件。你可以从Poppler for Windows下载并将其bin目录添加到系统PATH环境变量中。

2. 利用OpenCV进行图像分析：识别饼图切片

一旦PDF页面被转换为图像，我们就可以使用图像处理库（如OpenCV）来识别和分析饼图的视觉元素。核心思路是通过检测图像中的轮廓来识别饼图的各个切片。

安装OpenCV：

pip install opencv-python

识别饼图切片示例：

易标AI

告别低效手工，迎接AI标书新时代！3分钟智能生成，行业唯一具备查重功能，自动避雷废标项

135 查看详情易标AI

下面的代码片段演示了如何加载图像，将其转换为灰度图，进行阈值处理以突出饼图区域，然后查找并绘制轮廓。每个检测到的轮廓理论上可以代表饼图的一个切片。

import cv2
import numpy as np

def analyze_pie_chart_slices(image_path):
    """
    加载图像，识别饼图切片，并显示结果。
    :param image_path: 包含饼图的图像路径。
    """
    # 1. 加载图像
    image = cv2.imread(image_path)
    if image is None:
        print(f"错误：无法加载图像 '{image_path}'。请检查路径。")
        return

    # 2. 转换为灰度图
    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

    # 3. 阈值化处理
    # 使用二值反转阈值，将饼图切片（通常颜色较深）变为白色，背景变为黑色
    # 阈值128是一个经验值，可能需要根据具体图像调整
    _, thresh = cv2.threshold(gray, 128, 255, cv2.THRESH_BINARY_INV)

    # 4. 查找轮廓
    # RETR_EXTERNAL 仅检测最外层轮廓，CHAIN_APPROX_SIMPLE 压缩水平、垂直和对角线段
    contours, _ = cv2.findContours(thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

    print(f"在图像 '{image_path}' 中检测到 {len(contours)} 个可能的切片轮廓。")

    # 5. 绘制轮廓（可选：用于可视化）
    # 在原始图像上绘制所有检测到的轮廓，颜色为绿色 (0, 255, 0)，线条粗细为2
    image_with_contours = image.copy()
    cv2.drawContours(image_with_contours, contours, -1, (0, 255, 0), 2)

    # 6. 显示结果
    cv2.imshow('原始图像', image)
    cv2.imshow('灰度图', gray)
    cv2.imshow('阈值图', thresh)
    cv2.imshow('带有轮廓的图像', image_with_contours)
    cv2.waitKey(0) # 等待按键，然后关闭所有窗口
    cv2.destroyAllWindows()

    return contours, image # 返回轮廓和原始图像，以便进一步处理

# 示例用法
# 假设我们已经将PDF转换为图像，并获取了第一页的图像路径
# if image_files:
#     pie_chart_image_path = image_files[0] # 假设饼图在第一页
#     detected_contours, original_image = analyze_pie_chart_slices(pie_chart_image_path)

3. 从轮廓中提取数据：量化饼图切片

仅仅识别出轮廓是不够的，我们的目标是提取每个切片所代表的数据（例如，百分比或数值）。这通常需要以下几个步骤：

3.1 计算切片面积或角度

对于饼图，每个切片的大小与其所占的比例成正比。我们可以计算每个轮廓的面积，然后将其与所有切片总面积进行比较，从而得出百分比。

def calculate_slice_percentages(contours, total_chart_area=None):
    """
    计算每个饼图切片的面积，并估算其在总面积中的百分比。
    :param contours: 饼图切片的OpenCV轮廓列表。
    :param total_chart_area: 如果已知整个饼图的面积，可以提供，否则会从所有切片面积之和计算。
    :return: 包含每个切片面积和百分比的列表。
    """
    slice_data = []

    # 计算每个切片的面积
    areas = [cv2.contourArea(c) for c in contours]

    # 如果未提供总图表面积，则假定为所有切片面积之和
    if total_chart_area is None:
        total_chart_area = sum(areas)

    if total_chart_area == 0:
        print("警告：总图表面积为零，无法计算百分比。")
        return []

    for i, area in enumerate(areas):
        percentage = (area / total_chart_area) * 100
        slice_data.append({
            'slice_index': i,
            'area': area,
            'percentage': percentage
        })
    return slice_data

# 示例：
# if 'detected_contours' in locals() and detected_contours:
#     slice_percentages = calculate_slice_percentages(detected_contours)
#     for data in slice_percentages:
#         print(f"切片 {data['slice_index']}: 面积={data['area']:.2f}, 占比={data['percentage']:.2f}%")

3.2 颜色分析与映射

如果饼图的每个切片都用不同的颜色表示不同的类别，我们可以提取每个切片区域的平均颜色，并尝试将其映射到预定义的颜色-类别字典。

def get_*erage_color(image, contour):
    """
    获取给定轮廓区域内的平均颜色。
    :param image: 原始图像。
    :param contour: OpenCV轮廓。
    :return: BGR格式的平均颜色元组。
    """
    mask = np.zeros(image.shape[:2], dtype="uint8")
    cv2.drawContours(mask, [contour], -1, 255, -1) # 填充轮廓区域
    mean_color = cv2.mean(image, mask=mask)[:3] # 获取BGR通道的平均值
    return tuple(map(int, mean_color))

def map_colors_to_categories(slice_data, original_image, contours, color_category_map=None):
    """
    为每个切片添加平均颜色信息，并尝试映射到类别。
    :param slice_data: 包含切片面积和百分比的列表。
    :param original_image: 原始彩色图像。
    :param contours: 饼图切片的OpenCV轮廓列表。
    :param color_category_map: 一个字典，键为颜色元组，值为类别字符串。
    :return: 更新后的切片数据列表。
    """
    updated_slice_data = []
    for i, data in enumerate(slice_data):
        contour = contours[i]
        *g_color = get_*erage_color(original_image, contour)
        data['*erage_color_bgr'] = *g_color

        if color_category_map:
            # 简单匹配，实际可能需要更复杂的颜色相似度算法
            matched_category = None
            for color_key, category_name in color_category_map.items():
                # 假设颜色完全匹配，实际需考虑颜色容差
                if *g_color == color_key: 
                    matched_category = category_name
                    break
            data['category'] = matched_category if matched_category else "未知类别"
        updated_slice_data.append(data)
    return updated_slice_data

# 示例：
# 假设我们有一个颜色到类别的映射
# color_map = {
#     (100, 100, 200): "类别A", # 示例颜色 (B, G, R)
#     (200, 100, 100): "类别B",
#     (100, 200, 100): "类别C"
# }
# if 'slice_percentages' in locals() and 'original_image' in locals():
#     final_slice_data = map_colors_to_categories(slice_percentages, original_image, detected_contours, color_map)
#     for data in final_slice_data:
#         print(f"切片 {data['slice_index']}: 颜色={data['*erage_color_bgr']}, 类别={data.get('category', 'N/A')}, 占比={data['percentage']:.2f}%")

3.3 文本识别（OCR）

如果饼图带有标签文本，可以使用OCR（光学字符识别）技术来提取这些文本。pytesseract是Python中一个流行的OCR库，它封装了Google Tesseract OCR引擎。

安装pytesseract：

pip install pytesseract

注意事项： pytesseract同样需要安装Tesseract OCR引擎的二进制文件。

import pytesseract

def extract_text_from_region(image, region_coords):
    """
    从图像的指定区域提取文本。
    :param image: 原始图像。
    :param region_coords: 区域坐标 (x, y, w, h)。
    :return: 提取到的文本。
    """
    x, y, w, h = region_coords
    roi = image[y:y+h, x:x+w]
    text = pytesseract.image_to_string(roi, lang='eng') # 指定语言，例如 'eng'
    return text.strip()

# 示例：
# 假设我们知道某个切片旁边标签的近似坐标
# label_region = (x, y, w, h) # 需要手动或通过其他图像处理方法确定
# if 'original_image' in locals():
#     extracted_label = extract_text_from_region(original_image, label_region)
#     print(f"提取到的标签文本: {extracted_label}")

OCR的准确性高度依赖于图像质量和文本样式。对于图表中的小字体或特殊字体，可能需要进行图像预处理（如增强对比度、去噪）以提高识别率。

4. 注意事项与优化

PDF类型： 矢量PDF（文本和图形是可编辑的对象）比扫描件PDF（本质上是图像）更容易处理。对于矢量PDF，有时可以直接解析其内部的绘图指令来获取数据，但这种方法通常更复杂且缺乏通用性。本文的图像处理方法适用于两种类型的PDF。
图表复杂性： 饼图相对简单。对于柱状图、折线图等，需要不同的图像处理策略来识别柱子、线条和轴刻度。例如，柱状图可能需要识别矩形，折线图可能需要识别线条段和交点。
图像预处理： 在进行轮廓检测或OCR之前，对图像进行适当的预处理（如高斯模糊去噪、自适应阈值、形态学操作）可以显著提高结果的准确性。
颜色识别精度： 简单的颜色匹配可能不准确。更高级的方法包括计算颜色直方图、使用颜色空间转换（如HSV）或K-Means聚类来识别主导颜色。
交互式调试： 在开发过程中，利用cv2.imshow()函数实时显示中间处理结果（如灰度图、阈值图、轮廓图）对于调试和优化参数至关重要。

总结

通过结合pdf2image和OpenCV等Python库，我们可以构建一个强大的系统，用于从PDF文档中提取饼图数据。整个流程包括将PDF转换为图像、使用图像处理技术识别饼图切片、以及通过面积计算、颜色分析或OCR进一步量化和识别数据。虽然这需要一定的图像处理知识，但它为自动化数据提取提供了一个灵活且通用的解决方案，尤其适用于那些传统PDF解析工具难以处理的复杂图表。在实际应用中，可能需要根据具体的PDF和图表样式对参数和算法进行细致的调整和优化。