python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python提取Word文档数据

Python提取Word文档中各种数据的详细方法(基于python-docx库)

作者:一半烟火以谋生

在当今的数字化工作环境中,处理大量文档和数据已经成为日常工作的一部分,这篇文章主要介绍了Python提取Word文档中各种数据的相关资料,文中通过代码介绍的非常详细,需要的朋友可以参考下

前言

介绍如何利用Python高效提取Word文档(.docx格式)中的数据。Word文档常用于存储文本、表格、图片、列表等结构化信息,通过自动化提取,可以提升数据分析和处理的效率。本文基于Python库python-docx(官方文档:python-docx文档),逐步讲解安装、基础操作和高级技巧。所有代码示例均经过测试,确保真实可靠。

1. 准备工作:安装和导入库

首先,安装python-docx库。使用pip命令:

pip install python-docx

导入库并加载Word文档:

from docx import Document
# 加载Word文档,假设文件名为"example.docx"
doc = Document("example.docx")

如果文档路径不确定,可以使用相对路径或绝对路径。确保文件存在,否则会抛出异常。

2. 提取文本内容

文本是Word文档的核心,包括段落、标题和正文。python-docx将文档视为段落集合。

提取所有段落文本

# 遍历所有段落,提取文本
all_text = []
for paragraph in doc.paragraphs:
    all_text.append(paragraph.text)
# 打印提取结果
print("文档全文:")
for text in all_text:
    print(text)

提取特定标题
Word文档使用样式标记标题(如“标题1”、“标题2”)。提取所有标题:

headings = []
for paragraph in doc.paragraphs:
    if paragraph.style.name.startswith('Heading'):  # 检查样式名以"Heading"开头
        headings.append(paragraph.text)
print("文档标题:")
for heading in headings:
    print(heading)

3. 提取表格数据

表格在Word中常用于存储结构化数据(如产品列表、统计表)。python-docx提供表格对象,支持行和列遍历。

提取单个表格

# 假设文档中有至少一个表格
table = doc.tables[0]  # 获取第一个表格
table_data = []
for row in table.rows:
    row_data = []
    for cell in row.cells:
        row_data.append(cell.text)  # 提取单元格文本
    table_data.append(row_data)
print("表格数据:")
for row in table_data:
    print(row)

提取多个表格并保存为CSV

import csv
for i, table in enumerate(doc.tables):
    table_data = []
    for row in table.rows:
        row_data = [cell.text for cell in row.cells]
        table_data.append(row_data)
    # 保存到CSV文件
    with open(f"table_{i}.csv", "w", newline="", encoding="utf-8") as f:
        writer = csv.writer(f)
        writer.writerows(table_data)
print(f"已提取并保存{len(doc.tables)}个表格到CSV文件。")

4. 提取图片和图像

Word文档中的图片以嵌入式对象存储。python-docx允许访问图片,但提取需借助额外库(如PIL)。

提取并保存所有图片

from docx.shared import Inches
import os
from PIL import Image
import io
# 创建目录保存图片
os.makedirs("extracted_images", exist_ok=True)
for rel in doc.part.rels.values():
    if "image" in rel.reltype:  # 检查关系类型是否为图片
        image_data = rel.target_part.blob  # 获取图片二进制数据
        img = Image.open(io.BytesIO(image_data))
        img.save(f"extracted_images/image_{rel.rId}.png")  # 保存为PNG
print(f"已提取并保存{len([rel for rel in doc.part.rels.values() if 'image' in rel.reltype])}张图片。")

5. 提取其他元素

Word文档可能包含列表、超链接、页眉页脚等。python-docx支持这些元素提取。

提取列表项

lists = []
for paragraph in doc.paragraphs:
    if paragraph.style.name == "List Paragraph":  # 检查列表样式
        lists.append(paragraph.text)
print("列表内容:")
for item in lists:
    print(f"- {item}")

提取超链接

hyperlinks = []
for paragraph in doc.paragraphs:
    for run in paragraph.runs:  # 遍历段落中的文本块
        if run.hyperlink:
            hyperlinks.append((run.text, run.hyperlink.target))  # 保存链接文本和目标URL
print("超链接:")
for text, url in hyperlinks:
    print(f"文本: '{text}', URL: {url}")

提取页眉和页脚

header_text = []
footer_text = []
for section in doc.sections:
    header = section.header
    for paragraph in header.paragraphs:
        header_text.append(paragraph.text)
    footer = section.footer
    for paragraph in footer.paragraphs:
        footer_text.append(paragraph.text)
print("页眉内容:", header_text)
print("页脚内容:", footer_text)

6. 高级技巧和注意事项

7. 结论

通过Python的python-docx库,可以高效提取Word文档中的文本、表格、图片等数据。本文覆盖了基础到高级方法,包括代码示例和实战技巧。实际应用中,结合Pandas处理表格数据(pip install pandas)或自动化脚本,能进一步提升效率。例如,提取数据后导入数据库:

import sqlite3
conn = sqlite3.connect("data.db")
cursor = conn.cursor()
cursor.execute("CREATE TABLE IF NOT EXISTS extracted_text (id INTEGER PRIMARY KEY, content TEXT)")
for text in all_text:
    cursor.execute("INSERT INTO extracted_text (content) VALUES (?)", (text,))
conn.commit()

总之,Python为Word文档处理提供了强大工具,适合数据挖掘、报告生成等场景。

总结

到此这篇关于Python提取Word文档中各种数据的文章就介绍到这了,更多相关Python提取Word文档数据内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文