当前位置: 代码网 > it编程>前端脚本>Python > 使用Python实现网页表格转换为markdown

使用Python实现网页表格转换为markdown

2025年05月28日 Python 我要评论
在日常工作中,我们经常需要从网页上复制表格数据,并将其转换成markdown格式,以便在文档、邮件或论坛中使用。然而,手动转换不仅耗时,还容易出错。今天,就为大家带来一款网页表格转markdown的利

在日常工作中,我们经常需要从网页上复制表格数据,并将其转换成markdown格式,以便在文档、邮件或论坛中使用。然而,手动转换不仅耗时,还容易出错。今天,就为大家带来一款网页表格转markdown的利器,帮你一键完成转换,轻松应对各种场景!

场景需求

想象一下,你需要从公司内部网站复制一份销售数据表,并将其发送给团队成员。你希望他们能够轻松查看和编辑这份数据,但直接复制粘贴往往格式错乱。这时,如果有一个工具能将表格自动转换为markdown格式,那该有多好!

解决方案

本文介绍的python脚本,正是为解决这一问题而生。它利用requests-html库抓取网页内容,并通过自定义函数table_to_markdown将html表格转换为markdown格式。不仅如此,脚本还能处理复杂的单元格合并,确保转换后的表格美观、准确。

核心功能

  • 自动提取链接:脚本会识别并保留表格中的超链接,转换成markdown格式的链接。
  • 合并单元格支持:即使是跨越多行或多列的单元格,也能正确转换。
  • 异常处理:在提取过程中遇到任何异常,都会友好提示,保证程序的健壮性。

安装依赖

确保你已经安装了python及以下库:

pip install requests-html pyperclip

使用方法

运行脚本,输入目标网页的url。

脚本会自动抓取网页上的第一个表格。

转换后的markdown表格将显示在终端,并自动复制到剪贴板。

示例代码

from rich import print  
from rich.progress import track
from rich.console import console
from rich.logging import richhandler
import logging
from requests_html import htmlsession
import pyperclip
from collections import defaultdict
import sys

# 初始化日志
console = console()
logging.basicconfig(
    level="info", format="%(message)s", datefmt="[%x]", handlers=[richhandler(console=console)]
)
log = logging.getlogger("rich")


def extract_links(element):
    """安全处理链接提取,避免索引越界"""
    if not element.html:
        return ""
    
    # 直接处理元素内容,不依赖body标签
    text_parts = []
    processed_anchors = set()
    
    # 先处理所有<a>标签
    for a in element.find('a'):
        href = a.attrs.get('href', '')
        if href and a.text:
            text_parts.append(f"[{a.text.strip()}]({href})")
            # 标记这个a标签的文本节点已处理
            processed_anchors.add(a.text.strip())
    
    # 添加非链接文本(且未被链接包含的文本)
    full_text = element.text
    for anchor_text in processed_anchors:
        full_text = full_text.replace(anchor_text, '')
    
    if full_text.strip():
        text_parts.insert(0, full_text.strip())
    
    return ' '.join(text_parts).strip()

def table_to_markdown(table):
    """处理合并单元格的表格转换"""
    rows = table.find('tr')
    if not rows:
        return ""

    # 初始化数据结构
    table_grid = []
    rowspan_tracker = defaultdict(dict)  # {row: {col: (content, remaining_span)}}
    max_cols = 0

    for row_idx, row in enumerate(rows):
        cells = row.find('th, td')
        col_idx = 0
        current_row = []

        # 处理活跃的跨行单元格
        while col_idx in rowspan_tracker.get(row_idx, {}):
            content, remaining = rowspan_tracker[row_idx][col_idx]
            current_row.append(content)
            if remaining > 1:
                rowspan_tracker[row_idx + 1][col_idx] = (content, remaining - 1)
            col_idx += 1

        # 处理当前单元格
        for cell in cells:
            # 跳过已填充位置
            while col_idx < len(current_row) and current_row[col_idx] is not none:
                col_idx += 1

            # 解析单元格(处理链接和合并属性)
            content = extract_links(cell)
            colspan = int(cell.attrs.get('colspan', 1))
            rowspan = int(cell.attrs.get('rowspan', 1))

            # 主单元格
            current_row.append(content)
            
            # 列合并处理(复制内容)
            for _ in range(1, colspan):
                current_row.append(content)
            
            # 行合并处理
            if rowspan > 1:
                for r in range(1, rowspan):
                    if row_idx + r not in rowspan_tracker:
                        rowspan_tracker[row_idx + r] = {}
                    rowspan_tracker[row_idx + r][col_idx] = (content, rowspan - r)
            
            col_idx += colspan

        # 填充空白
        current_row = [cell if cell is not none else "" for cell in current_row]
        table_grid.append(current_row)
        max_cols = max(max_cols, len(current_row))

    # 统一列宽
    for row in table_grid:
        row.extend([""] * (max_cols - len(row)))

    # 生成markdown
    markdown = []
    if table_grid:
        # 表头
        markdown.append("| " + " | ".join(table_grid[0]) + " |")
        markdown.append("| " + " | ".join(["---"] * len(table_grid[0])) + " |")
        
        # 表格内容
        for row in table_grid[1:]:
            markdown.append("| " + " | ".join(row) + " |")

    return "\n".join(markdown)

def get_table_as_markdown(url, table_index=0, timeout=20):
    try:
        session = htmlsession()
        response = session.get(url)
        response.html.render(timeout=timeout)
        
        tables = response.html.find('table')
        if not tables:
            return "未找到表格"
            
        if table_index >= len(tables):
            return f"表格索引超出范围(共 {len(tables)} 个表格)"
            
        return table_to_markdown(tables[table_index])
        
    except exception as e:
        return f"错误: {str(e)}"

if __name__ == "__main__":
    url = input("请输入网页url: ")
    result = get_table_as_markdown(url)
    
    print("\n生成的markdown表格:\n")
    print(result)
    
    try:
        pyperclip.copy(result)
        print("\n✅ 已复制到剪贴板")
    except:
        print("\n⚠️ 无法复制到剪贴板,请手动复制")

结语

这款脚本不仅能大幅提升你的工作效率,还能确保表格格式的准确性和一致性。无论是日常办公还是学术研究,它都是你不可或缺的好帮手。赶快试试吧,让你的数据处理工作变得简单又高效。

到此这篇关于使用python实现网页表格转换为markdown的文章就介绍到这了,更多相关python网页转markdown内容请搜索代码网以前的文章或继续浏览下面的相关文章希望大家以后多多支持代码网!

(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com