用Python轻松玩转数据可视化——将数据生成HTML并保存为图片

2023-05-16 07:55:38

使用 Python 将数据可视化：从 HTML 表格到美观图片

数据可视化在当今数据驱动世界中的重要性

在数据无处不在的时代，将数据清晰且引人注目地呈现变得至关重要。数据可视化通过以表格或图形的形式展示数据，极大地提高了数据的可读性和理解性。Python，作为一门强大的编程语言，为数据可视化提供了丰富的库和工具，让您能够轻松创建令人印象深刻的数据可视化效果。

使用 Python 创建 HTML 表格

准备工作

在开始之前，您需要：

安装 Python 3.x
一个文本编辑器或 IDE
一个文件夹用于保存 HTML 文件

数据准备

首先，将您的数据组织成一个 Python 字典。例如：

data = {
    "姓名": ["张三", "李四", "王五"],
    "年龄": [20, 25, 30],
    "性别": ["男", "女", "男"]
}

生成 HTML 表格

使用 Python 的 html 库创建 HTML 表格：

from html import escape
html = """
<!DOCTYPE html>
<html>
<head></head>
<body>
<table>
"""

# 添加表头
html += "<tr>"
for key in data:
    html += f"<th>{escape(key)}</th>"
html += "</tr>"

# 添加数据行
for row in zip(*data.values()):
    html += "<tr>"
    for value in row:
        html += f"<td>{escape(value)}</td>"
    html += "</tr>"

html += """
</table>
</body>
</html>
"""

with open("table.html", "w") as f:
    f.write(html)

使用 CSS 美化 HTML 表格

创建 CSS 文件

创建 style.css 文件并添加样式：

table {
    border-collapse: collapse;
    width: 100%;
}

th, td {
    border: 1px solid black;
    padding: 5px;
    text-align: center;
}

th {
    background-color: #eee;
}

将 CSS 添加到 HTML 文件

在 HTML 文件中添加 <link> 标签：

<head>
<link rel="stylesheet" href="style.css">
</head>

将 HTML 表格保存为图片

使用 wkhtmltopdf 库将 HTML 转换为 PDF：

import pdfkit
pdfkit.from_file("table.html", "table.pdf")

使用 ImageMagick 将 PDF 转换为 PNG：

import os
from PIL import Image
im = Image.open("table.pdf")
im.save("table.png")
os.remove("table.pdf")

示例代码

完整的示例代码：

from html import escape
import pdfkit
from PIL import Image
import os

data = {
    "姓名": ["张三", "李四", "王五"],
    "年龄": [20, 25, 30],
    "性别": ["男", "女", "男"]
}

html = """
<!DOCTYPE html>
<html>
<head>
<link rel="stylesheet" href="style.css">
</head>
<body>
<table>
"""

# 添加表头
html += "<tr>"
for key in data:
    html += f"<th>{escape(key)}</th>"
html += "</tr>"

# 添加数据行
for row in zip(*data.values()):
    html += "<tr>"
    for value in row:
        html += f"<td>{escape(value)}</td>"
    html += "</tr>"

html += """
</table>
</body>
</html>
"""

with open("table.html", "w") as f:
    f.write(html)

pdfkit.from_file("table.html", "table.pdf")
im = Image.open("table.pdf")
im.save("table.png")
os.remove("table.pdf")