优秀的编程知识分享平台

网站首页 > 技术文章 正文

Python 中的数据可视化:将列表转换为图形

nanyue 2024-10-18 07:39:11 技术文章 9 ℃

我们经常使用 Microsoft Excel和 Microsoft Word 等应用程序轻松地绘制柱形图。如果我们想更好地控制图形创建过程,以编程方式创建此类图形将是最佳选择。

在本文中,我将向您展示如何使用 Python 创建柱形图。

使用 Pillow

要使用 Pillow 绘制柱形图,我们主要需要两个模块:Image 和 ImageDraw 模块,这两个模块都将从 Pillow (PIL) 导入。Image 模块将用于加载图像,而 ImageDraw 模块将用于创建 2D 图形。

「安装 Pillow」

pip install Pillow

现在,让我们逐步介绍程序的创建过程。

首先,我们需要一张空白图像 “blank.png”,我们将在其上绘制条形图。让我们加载这张空白图像准备绘制柱形图。

from PIL import Image, ImageDraw
img = Image.open('blank.png')
draw_img = ImageDraw.Draw(img)

我们使用列表,准备绘制所需的数据。

data = [15, 25, 55, 10, 100, 60]

然后,我们利用这些数据来绘制柱形图。我们可以把图表上的柱形视为线。因此,我们将使用 line() 方法来完成。

for i in data:
    x = x + 30  
    y = 200 - int(i) 
    draw_img.line((x,200,x,y), width=10, fill=(255,0,0,255))

以上代码遍历列表,并使用列表中的数据绘制一个柱形图。需要特别注意的是,坐标轴的原点(0,0)位于图像的左上角。x = x + 30 用于限定间距。width=10 限定了线宽。fill=(255,0,0,255) 设置颜色。循环遍历完成后,利用列表数据,绘制了从(30,200)到(30,185),从(60,200)到(60,175),从(90,200)到(90,145),从(120,200)到(120,190),从(150,200)到(150,100),从(180,200)到(180,140)等多条线段。

最后,我们可以使用以下语句查看图像:img.show()

完整代码:

from PIL import Image, ImageDraw
img = Image.open('blank.png')
draw_img = ImageDraw.Draw(img)
data = [15, 25, 55, 10, 100, 60]
x = 0
for i in data:
    x = x + 30  
    y = 200 - int(i) 
    draw_img.line((x,200,x,y), width=10, fill=(255,0,0,255))
img.show()

输出柱形图:

使用 Matplotlib

Matplotlib 是一个可视化库,可以轻松在 Python 中创建可视化图形。

「安装 Matplotlib。」

pip install matplotlib

导入 Matplotlib 的子模块 pyplotplt。

import matplotlib.pyplot as plt

我们将数据转换为整数。

data = list(map(int, data))

利用列表长度,设置坐标 x 。

x = range(len(data))

然后,使用 Matplotlib 的 bar() 函数绘制柱形图,并显示该图形。

plt.bar(x,data)
plt.show()

完整代码:

import matplotlib.pyplot as plt
data = [15, 25, 55, 10, 100, 60]
data = list(map(int, data))
x = range(len(data))
plt.bar(x,data)
plt.show()

输出柱形图:

?

文章创作不易,如果您喜欢这篇文章,请关注、点赞并分享给朋友。如有意见和建议,请在评论中反馈。

?

最近发表
标签列表