如何使用 matplotlib 在 Python 3 中绘制数据

前些天发现了一个巨牛的人工智能学习网站,通俗易懂,风趣幽默,忍不住分享一下给大家。点击跳转到网站

简介

Python 在处理数据方面非常出色。通常,数据集会包括多个变量和许多实例,这使得很难理解数据的情况。数据可视化是帮助您识别数据模式的一种有用方式。

例如,假设您是一名房地产经纪人,您想要了解房屋的年龄与售价之间的关系。如果您的数据包括 5 栋房屋的数据,那么理解情况就不会太困难。但是,假设您想要使用整个镇上 500 栋房屋的数据,那么理解年龄如何影响价格就会变得非常困难。通过绘制售价与年龄的关系,对数据进行可视化肯定会阐明这两者之间的关系。

可视化是以一种通用方式快速传达概念的方法,特别是对于那些不熟悉您数据的人。每当我们处理数据时,可视化通常是分析的必要部分。

我们将使用 2D 绘图库 matplotlib,最初由 John D. Hunter 编写,自那以后已成为一个非常活跃的开源开发社区项目。它允许您生成高质量的折线图、散点图、直方图、条形图等。每个图以不同的方式呈现数据,通常在确定数据的最具信息量的图之前,尝试不同类型的图表是很有用的。要记住,可视化是艺术和科学的结合。

鉴于可视化的重要性,本教程将描述如何使用 matplotlib 在 Python 中绘制数据。我们将介绍如何使用一小组数据生成散点图,向图表添加标题和图例,并通过更改绘图点的外观来自定义图表。

完成本教程后,您将能够在 Python 中绘制数据!

先决条件

在进行本教程之前,您应该已经安装了 Python 3,并在计算机上设置了本地编程环境。如果不是这种情况,您可以按照适用于您操作系统的相应安装和设置指南进行设置。

步骤 1 --- 导入 matplotlib

在我们开始在 Python 中工作之前,让我们再次检查 matplotlib 模块是否已安装。在命令行中,通过运行以下命令检查 matplotlib 是否已安装:

command 复制代码
python -c "import matplotlib"

如果 matplotlib 已安装,此命令将顺利完成,我们可以开始了。如果没有安装,您将收到错误消息:

command 复制代码
Traceback (most recent call last): File "<string>", line 1, in <module> ImportError: No module named 'matplolib'

如果收到错误消息,请使用 pip 下载该库:

command 复制代码
pip install matplotlib

现在 matplotlib 已安装,我们可以在 Python 中导入它。首先,让我们创建我们将在本教程中使用的脚本:scatter.py。然后,在我们的脚本中,让我们导入 matplotlib。由于我们只会使用绘图模块(pyplot),因此在导入时让我们指定它。

python 复制代码
import matplotlib.pyplot as plt

我们通过在 matplotlib 末尾添加 .pyplot 来指定我们希望导入的模块。为了在脚本中更容易地引用该模块,我们将其缩写为 plt。现在,我们可以继续创建和绘制我们的数据。

步骤 2 --- 创建要绘制的数据点

在我们的 Python 脚本中,让我们创建一些要处理的数据。我们将使用 2D 数据,因此我们将需要每个数据点的 X 和 Y 坐标。

为了更好地理解 matplotlib 的工作原理,我们将把我们的数据与可能的现实场景联系起来。假设我们是一家咖啡店的老板,我们对一年中的平均天气和冰咖啡的总销量之间的关系感兴趣。我们的 X 变量将是每月销售的冰咖啡总数,我们的 Y 变量将是每个月的华氏平均温度。

在我们的 Python 脚本中,我们将创建两个列表变量:X(冰咖啡总销量)和 Y(平均温度)。我们各自列表中的每个项目将代表每个月的数据(一月到十二月)。例如,在一月份,平均温度为 32 华氏度,咖啡店卖出了 590 杯冰咖啡。

python 复制代码
import matplotlib.pyplot as plt

X = [590,540,740,130,810,300,320,230,470,620,770,250]
Y = [32,36,39,52,61,72,77,75,68,57,48,48]

现在我们有了数据,我们可以开始绘图。

步骤 3 --- 绘制数据

散点图非常适合用于确定两个变量之间的关系,因此我们将使用这种图表类型作为示例。要使用 matplotlib 创建散点图,我们将使用 scatter() 函数。该函数需要两个参数,分别表示 X 和 Y 坐标值。

python 复制代码
import matplotlib.pyplot as plt

X = [590,540,740,130,810,300,320,230,470,620,770,250]
Y = [32,36,39,52,61,72,77,75,68,57,48,48]

plt.scatter(X,Y)
plt.show()

每次创建图表时,我们都必须使用 plt.show() 指定要显示图表。

在继续之前,让我们检查一下我们的脚本是否正常工作。保存脚本并通过命令行运行:

command 复制代码
python scatter.py

如果一切顺利,应该会启动一个窗口显示图表,如下所示:

!Alt 散点图

这个窗口非常适合查看数据;它是交互式的,并包括几个功能,比如悬停显示标签和坐标、放大或缩小以及保存。

步骤 4 --- 添加标题和标签

现在我们知道我们的脚本正常工作,我们可以开始向图表中添加信息。为了清楚地表明我们的数据代表什么,让我们在图表中包括一个标题以及每个轴的标签。

我们将从添加标题开始。我们在脚本的 plt.show() 行之前添加标题。

python 复制代码
import matplotlib.pyplot as plt

X = [590,540,740,130,810,300,320,230,470,620,770,250]
Y = [32,36,39,52,61,72,77,75,68,57,48,48]

plt.scatter(X,Y)

plt.title('温度与冰咖啡销量之间的关系')

plt.show()

接下来,在 plt.title 行下方添加轴的标签:

python 复制代码
...
plt.xlabel('售出的冰咖啡杯数')
plt.ylabel('华氏温度')
...

如果我们保存脚本并再次运行它,现在应该会得到一个更新的图表,更具信息性。我们的更新后的图表应该看起来像这样:

!Alt 带有标题和 X/Y 标签的散点图

步骤 5 --- 自定义图表

我们处理的每个数据集都是独特的,能够自定义我们希望如何显示信息非常重要。记住,可视化也是一门艺术,所以可以在其中发挥创意!matplotlib 包括许多自定义功能,比如不同的颜色、点符号和大小。根据我们的需求,我们可能想尝试不同的比例,为我们的轴使用不同的范围。我们可以通过指定轴的新范围来更改默认参数,如下所示:

python 复制代码
import matplotlib.pyplot as plt

X = [590,540,740,130,810,300,320,230,470,620,770,250]
Y = [32,36,39,52,61,72,77,75,68,57,48,48]

plt.scatter(X,Y)

plt.xlim(0,1000)
plt.ylim(0,100)

plt.title('温度与冰咖啡销量之间的关系')

plt.show()
...

原始图表中的点看起来有点小,蓝色可能不是我们想要的颜色。也许我们想要三角形而不是圆圈作为点的符号。如果我们想要更改点的实际颜色/大小/形状,我们必须在初始的 plt.scatter() 调用中进行这些更改。我们将更改以下参数:

  • s:点的大小,默认值为 20
  • c:颜色、序列或颜色序列,默认值为 'b'
  • marker:点符号,默认值为 'o'

可能的标记包括许多不同的形状,如菱形、六边形、星形等。颜色选择包括蓝色、绿色、红色和洋红色等。还可以提供 HTML 十六进制字符串作为颜色。请参阅 matplotlib 的文档,了解可能的标记和颜色的全面列表。

为了使我们的图表更易于阅读,让我们将点的大小加倍(s=60),将颜色更改为红色(c='r'),并将符号更改为三角形(marker='^')。我们将修改 plt.scatter() 函数:

python 复制代码
plt.scatter(X, Y, s=60, c='red', marker='^')

在运行更新后的脚本之前,我们可以再次检查我们的代码是否正确。自定义图表的更新脚本应该看起来像这样:

python 复制代码
import matplotlib.pyplot as plt

X = [590,540,740,130,810,300,320,230,470,620,770,250]
Y = [32,36,39,52,61,72,77,75,68,57,48,48]

#散点图

python 复制代码
plt.scatter(X, Y, s=60, c='red', marker='^')

#更改坐标轴范围

python 复制代码
plt.xlim(0,1000)
plt.ylim(0,100)

#添加标题

python 复制代码
plt.title('温度与冰咖啡销量的关系')

#添加 x 和 y 轴标签

python 复制代码
plt.xlabel('售出冰咖啡杯数')
plt.ylabel('华氏温度')

#显示图表

python 复制代码
plt.show()

不要忘记在进行第 6 步之前保存你的脚本。

第 6 步 --- 保存图表

现在我们已经完成了代码,让我们运行它来查看我们新定制的图表。

command 复制代码
python scatter.py

现在应该会打开一个窗口显示我们的图表:

!Alt 带有标题和 X/Y 标签的最终散点图,并使用更大、红色、三角形点进行定制。

接下来,通过单击保存按钮(位于底部工具栏上的磁盘图标)来保存图表。请记住,图像将被保存为 PNG 格式,而不是交互式图表。恭喜你,现在你拥有自己定制的散点图!

结论

在本教程中,你学会了如何使用 Python 中的 matplotlib 绘制数据。现在你可以可视化数据并定制图表。

要继续练习 matplotlib,你可以参考我们的指南"如何使用 Python 3 和 matplotlib 绘制词频图"。

相关推荐
ZJ_.几秒前
WPSJS:让 WPS 办公与 JavaScript 完美联动
开发语言·前端·javascript·vscode·ecmascript·wps
Narutolxy6 分钟前
深入探讨 Go 中的高级表单验证与翻译:Gin 与 Validator 的实践之道20241223
开发语言·golang·gin
Hello.Reader13 分钟前
全面解析 Golang Gin 框架
开发语言·golang·gin
禁默24 分钟前
深入浅出:AWT的基本组件及其应用
java·开发语言·界面编程
Cachel wood31 分钟前
python round四舍五入和decimal库精确四舍五入
java·linux·前端·数据库·vue.js·python·前端框架
Code哈哈笑34 分钟前
【Java 学习】深度剖析Java多态:从向上转型到向下转型,解锁动态绑定的奥秘,让代码更优雅灵活
java·开发语言·学习
終不似少年遊*36 分钟前
pyecharts
python·信息可视化·数据分析·学习笔记·pyecharts·使用技巧
程序猿进阶37 分钟前
深入解析 Spring WebFlux:原理与应用
java·开发语言·后端·spring·面试·架构·springboot
Python之栈38 分钟前
【无标题】
数据库·python·mysql
qq_4336184439 分钟前
shell 编程(二)
开发语言·bash·shell