Python使用Mermaid與Graphviz實現(xiàn)關系圖表生成的實戰(zhàn)指南

更新時間：2026年01月19日 08:17:48 作者：weixin_46244623

這篇文章主要為大家詳細介紹了兩種Python生成關系圖表的方法,即Mermaid和Graphviz,下面小編就和大家詳細介紹一下具體的實現(xiàn)方法吧

一、準備工作

1. 安裝依賴

pip install pydantic graphviz

注意：Graphviz 生成圖像需要系統(tǒng)安裝 Graphviz 可執(zhí)行文件。

Ubuntu/Debian：

sudo apt-get install graphviz

macOS：

brew install graphviz

Windows：請從 Graphviz 官網(wǎng) 下載并安裝。

二、定義關系模型

我們使用 pydantic 定義實體關系模型，每條關系包含三個屬性：subject、predicate、object。

from pydantic import BaseModel, Field

class Relationship(BaseModel):
    subject: str = Field(...)
    predicate: str = Field(...)
    object: str = Field(...)

三、生成 Mermaid 圖表

Mermaid 圖表是文本驅動的，我們可以根據(jù)關系列表自動生成 Mermaid 流程圖代碼。

from graphviz_render import relationships_to_mermaid

# 示例關系
relationships = [
    Relationship(subject="用戶", predicate="登錄", object="系統(tǒng)"),
    Relationship(subject="系統(tǒng)", predicate="驗證", object="身份"),
    Relationship(subject="數(shù)據(jù)庫", predicate="存儲", object="用戶信息"),
    Relationship(subject="AI助手", predicate="處理", object="請求"),
    Relationship(subject="文檔", predicate="描述", object="功能")
]

mermaid_code = relationships_to_mermaid(relationships)
print("生成的Mermaid代碼:\n")
print(mermaid_code)

Mermaid 輸出示例

四、生成 Graphviz 圖表

Graphviz 可以直接生成圖片文件（PNG、SVG），適合嵌入文檔或網(wǎng)頁。

from pathlib import Path
from graphviz_render import relationships_to_png

output_path = Path("./test_relationships.png")

try:
    relationships_to_png(relationships, output_path)
    print(f"Graphviz圖表已保存到: {output_path.absolute()}")
except Exception as e:
    print(f"生成Graphviz圖表時出錯: {e}")
    print("可能原因：未安裝Graphviz或中文字體缺失")

運行后，你將得到如下關系圖：

用戶 --登錄--> 系統(tǒng)
系統(tǒng) --驗證--> 身份
數(shù)據(jù)庫 --存儲--> 用戶信息
AI助手 --處理--> 請求
文檔 --描述--> 功能

如果中文顯示異常，可以在 Graphviz 中設置字體，例如：

dot.attr(fontname="Microsoft YaHei")

五、完整測試腳本

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from pydantic import BaseModel, Field
from pathlib import Path
from graphviz_render import relationships_to_mermaid, relationships_to_png

class Relationship(BaseModel):
    subject: str = Field(...)
    predicate: str = Field(...)
    object: str = Field(...)

def test_mermaid_generation():
    relationships = [
        Relationship(subject="用戶", predicate="登錄", object="系統(tǒng)"),
        Relationship(subject="系統(tǒng)", predicate="驗證", object="身份"),
        Relationship(subject="數(shù)據(jù)庫", predicate="存儲", object="用戶信息"),
        Relationship(subject="AI助手", predicate="處理", object="請求"),
        Relationship(subject="文檔", predicate="描述", object="功能")
    ]
    mermaid_code = relationships_to_mermaid(relationships)
    print("生成的Mermaid代碼:\n", mermaid_code)
    return relationships

def test_graphviz_generation(relationships):
    output_path = Path("./test_relationships.png")
    try:
        relationships_to_png(relationships, output_path)
        print(f"Graphviz圖表已保存到: {output_path.absolute()}")
    except Exception as e:
        print(f"生成Graphviz圖表時出錯: {e}")

if __name__ == "__main__":
    rels = test_mermaid_generation()
    test_graphviz_generation(rels)