目录
首页 笔记 详情

Python matplotlib画图时图例说明(python描点画图小技巧)


Matplotlib 的简单使用

Matplotlib 是 Python 中一个非常实用的模块,使用 Matplotlib 可以绘制各式各样的图标。包括折线图、条形图、饼图、雷达图等。Matplotlib 是仿照 MATLAB 实现的,想要在 Python 中使用 Matplotlib 我们需要安装它,语句如下:

pip install matplotlib

下面我们用 Matplotlib 来实现一个简单的折线图:

import numpy as np
import matplotlib.pyplot as plt
# 准备 X 轴的数据
x = np.linspace(0, 10, 10)
# 准备 y 轴的数据
y = x**2
# 获取折线图
plt.plot(x, y)
# 显示折线图
plt.show()

在上面的代码中我们使用到了 NumPy 和 matplotlib.pyplot 库。其中 NumPy 是一个数学运算的库,它会在我们安装 Matplotlib 时安装。而 pyplot 是 Matplotlib 的子库,绘图的操作主要都在里面。

在绘制图表时,我们需要先准备 x 和 y 轴的数据。然后调用 plt.plot 函数绘制图表,最后调用 plt.show 函数显示图表,下面是我们绘制的图表:

Python 使用 Matplotlib 绘图详解

不过上面的图表过于单调,你无法从里面获取准确的信息,因此需要我们丰富一下图表。

绘制折线图并丰富信息

添加标题、x 轴、y 轴信息

在 plot 中提供了一个函数用于添加标题、x 轴和 y 轴的信息。我们用第一节的代码进行一下扩展:

import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(0, 10, 10)
y = x**2
plt.plot(x, y)
# 添加标题
plt.title("y = x^2")
# 添加 x 轴的信息
plt.xlabel("x")
# 添加 y 轴的信息
plt.ylabel("y")
# 保存图片
plt.savefig('1.jpg')
plt.show()

我们在调用 plt.plot 函数后又调用了三个函数,分别是 title、xlabel、ylabel。正好对应了我们的标题、x 轴和 y 轴。这样我们的图表就更容易理解,下面是绘制的图表:

Python 使用 Matplotlib 绘图详解

从图可以看出,它是 y = x^2 的函数图像。

现在有了一些提示信息,我们可以考虑美化一个图表了。

丰富样式

我们在前面的基础上继续修改:

import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(0, 10, 10)
y = x**2
# 使用背景
plt.style.use('ggplot')
# 丰富样式
plt.plot(x, y, c='r', marker='o', ls='-', )
plt.title("y = x^2")
plt.xlabel("x")
plt.ylabel("y")
plt.savefig('1.jpg')
plt.show()

在上面我们对两个部分进行了修改,首先我们在绘制图表前调用了:

plt.style.use('ggplot')

这个操作是使用 Matplotlib 中自带的样式,我们可以通过:

plt.style.available

来查看可选的样式。

第二部分是在绘制图表时添加了几个参数,它们的含义分别如下:

c	颜色
marker	点的样式
ls	线条样式

最后绘制出来的图表如下:

Python 使用 Matplotlib 绘图详解

可以看到在样式上有了很大的改观。当然,我们的 Matplotlib 还可以绘制其它图。

绘制散点图

散点图的绘制和折线图很像,我们直接来看代码:

import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(0, 10, 10)
y = x**2
plt.style.use('ggplot')
# 绘制散点图
plt.scatter(x, y, c='r', marker='o', ls='-', )
plt.title("y = x^2")
plt.xlabel("x")
plt.ylabel("y")
plt.savefig('1.jpg')
plt.show()

可以看到,我们的代码只做了非常简单的修改。我们把原本绘制折线图的函数 plt.plot 换成了 plt.scatter。其余代码直接照搬,这就是我们散点图的绘制,我们先看看绘制出来的图片:

Python 使用 Matplotlib 绘图详解

可以看到散点图正确的绘制出来了。我们可以尝试绘制其它样式的散点图:

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 10, 10)

y = x**2

plt.style.use('ggplot')

plt.scatter(x, y, c='b', marker='*')

plt.title("y = x^2")

plt.xlabel("x")

plt.ylabel("y")

plt.savefig('1.jpg')

plt.show()

上面我调整了一个标记颜色和点的样式,绘制出来的图表如下:

Python 使用 Matplotlib 绘图详解

绘制饼图和条形图

除了上面两种,Matplotlib 还可以绘制饼图和统计图。它们的绘制也非常简单:

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 10, 10)

y = [10, 3, 4, 7, 9, 3, 11, 23, 3, 17]

plt.rcParams['font.sans-serif'] = ['SimHei']

plt.style.use('ggplot')

plt.bar(x, y, color='b', tick_label=['01', '02', '03', '04', '05', '06', '07', '08', '09', '10'])

plt.title("一个测试统计图")

plt.xlabel("年份")

plt.ylabel("y 值")

plt.savefig('1.jpg')

plt.show()

这里我们多加了一些内容,我们来分别看看。首先第一个是:

plt.rcParams['font.sans-serif'] = ['SimHei']

这句代码的作用是让我们的图表可以显示中文,默认情况下我们显示中文是会乱码的。

第二部分,我们把绘制图表的函数换成了 plt.bar,而且参数也有些改变,含义如下:

color : 颜色
tick_label : x 轴的标签

后面的内容和之前一样,下面是我们图表的效果图:

Python 使用 Matplotlib 绘图详解

有了中文标题,我们就能更方便的表示图表了。

不过有时候我们一张图里面可能会包含两组数据,这个时候我们可以使用图例来进行区分:

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(10)

y1 = [10, 3, 4, 7, 9, 3, 11, 23, 3, 17]
y2 = [3, 4, 7, 9, 12, 16, 16, 17, 23, 17]

plt.rcParams['font.sans-serif'] = ['SimHei']

plt.style.use('ggplot')

bar_width = 0.35
tick_label = ['01', '02', '03', '04', '05', '06', '07', '08', '09', '10']

plt.bar(x, y1, bar_width, color='b', label='data1')
plt.bar(x+bar_width, y2, bar_width, color='r', label='data2')

plt.legend()

plt.xticks(x+bar_width/2, tick_label)

plt.title("一个测试统计图")

plt.xlabel("年份")

plt.ylabel("y 值")

plt.savefig('1.jpg')

plt.show()

这次我们为绘制了两次条形图,为了不让它们重合,我们把 x 轴添加了一个 bar_width。并且为每个条形图添加了一个 label 参数,这个参数就是会在图例中显示的内容。然后我们调用 legend 函数就能为图片添加图例了,下面是效果图:

Python 使用 Matplotlib 绘图详解

左上角显示的就是我们的图例信息。

接下来我们再看看饼图的绘制吧:

import matplotlib.pyplot as plt
data = [0.2, 0.1, 0.05, 0.25, 0.4]
colors = ['#ee0000', '#0000ee', '#777777', '#00ee00', '#eeee00']
kinds = ['data1', 'data2', 'data3', 'data4', 'data5']
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.style.use('ggplot')

plt.pie(data, labels=kinds, colors=colors)

plt.title("一个测试统计图")

plt.xlabel("年份")

plt.ylabel("y 值")

plt.savefig('1.jpg')

plt.show()

这里同样需要修改 plt 的绘制函数,这次使用的是 pie 函数。不同的图需要的参数也不同,这次我们需要的参数主要有:

x	: 数据
labels	: 每个数据的标签
colors	: 每个数据显示的颜色

下面是我们效果图:

Python 使用 Matplotlib 绘图详解

当然了,Matplotlib 中提供了图像远不止这些,更多的内容可以自己尝试一下,本次 Chat 就到这里了。


写笔记

咨询

您的电子邮箱地址不会被公开。