《深度学习入门》学习笔记二：matplotlib

文章目录

• • 1.绘制简单的图形
  • 2.pyplot的功能
  • 3.显示图片

1.绘制简单的图形

import numpy as np
import matplotlib.pyplot as plt

#生成数据
x = np.arange(0, 6, 0.1)
y = np.sin(x)

#绘制图形
plt.plot(x, y)
plt.show()

首先通过NumPy中的arange方法生成了从0到6，间隔为0.1的数据，设置成了x。并且对其进行sin计算，生成了相同数量的y，最终将x，y的数据传给了plt.plot绘制出图形，然后通过plt.show显示图形。如图所示。

2.pyplot的功能

pyplot当然也可以同时绘制多个图形，并且可以在图上添加注释。

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(0,6,0.1)
y1 = np.sin(x)
y2 = np.cos(x)

plt.plot(x, y1, label="sin")
plt.plot(x, y2, linestyle = "--", label="cos")
plt.xlabel("x")
plt.ylabel("y")
plt.title('sin & cos')
plt.legend()
plt.show()

效果如下图所示

在这段代码中，label代表的是线的标签，linestyle代表线的类型。可以见下表：

类型	简写	说明
solid（默认）	“-”	实线
dotted	“.”	点虚线
dashed	“–”	破折线
dashdot	“-.”	点划线
none	“”或者“ ”	不划线

xlabel和ylabel分别代表x轴标签和y轴标签，title代表标题。
更多详情参考：Matplotlib 教程

3.显示图片

pyplot还可以显示图像，通过imread函数。