深度学习--Matplotlib学习笔记(一)

时间:2024-11-20 07:33:22

深度学习中绘图工具必不可少,于是Matplotlib的强大功能便体现出来。下面讲解一下几种简单的图形的绘制方法。

绘制折线图

折线图是最简单的一类图形,下面我们用pylot实现绘制折线图的功能。

这个命令用来在jupyter内部显示图片

%matplotlib inline
  • 1

首先需要导入Matplotlib包中的Pyplot模块

import matplotlib.pyplot as plt
  • 1
'
运行

接下来,使用 NumPy 提供的函数 arange() 创建一组数据来绘制图像。

#引入numpy包
import numpy as np
#获得0到2π之间的ndarray对象
x = np.arange(0, math.pi*2, 0.05)
  • 1
  • 2
  • 3
  • 4

上述所得 x 的值作用到 x 轴上,而该值对应的正弦值,也就是 y 值,使用以下方法获取:

y = np.sin(x)
  • 1

使用 plot() 函数对 x、y 进行绘制。

plt.plot(x,y)
  • 1

主要的绘图工作已经完成,不过还需要绘制一些细节,需要我们补充一下,比如图像的标题(title)、x 轴与 y 轴的标签(label)等。

plt.xlabel("angle")
plt.ylabel("sine")
plt.title('sine wave')
  • 1
  • 2
  • 3

完整的程序代码如下:

    from matplotlib import pyplot as plt
    import numpy as np
    import math
    #调用方法弧度转为角度
    x = np.arange(0, math.pi*2, 0.05)
    y = np.sin(x)
    plt.plot(x,y)
    plt.xlabel("angle")
    plt.ylabel("sine")
    plt.title('sine wave')
    #使用show展示图像
    plt.show()
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12

代码执行后,显示结果如下:请添加图片描述
是不是很简单,下面再来绘制一个U-I特性图:

I(mA) 23.92 13.34 9.26 7.95 7.05 5.21 4.49
U(V) 4.78 5.32 5.53 5.59 5.64 5.71 5.76
%matplotlib inline
import torch
from IPython import display
from matplotlib import pyplot as plt
import numpy as np
import random
x=[5.00,4.17,3.57,3.33,3.12,2.78,2.50]
y=[0.00,0.83,1.43,1.67,1.88,2.22,2.50]
plt.plot(x,y)
plt.ylabel("U/V")
plt.xlabel("I/mA")
plt.title("U-I")
plt.show()
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13

请添加图片描述

绘制曲线图

PyLab 是一个面向 Matplotlib 的绘图库接口,其语法和 MATLAB 十分相近。它和 Pyplot 模快都够实现Matplotlib 的绘图功能。PyLab 是一个单独的模块,随 Matplotlib 软件包一起安装,该模块的导包方式和 Pyplot不同,如下所示:

    #Pyplot导包方式
    from matplotlib import pyplot as plt
    #PyLab导包有两种方式
    import pylab
    from pylab import *
  • 1
  • 2
  • 3
  • 4
  • 5

其实PyLab和Pyplot能够实现的功能基本一致,只是PyLab和MATLAB 十分相近,用法简便。

import numpy as np
x=np.linspace(-3, 3, 30)
y=x**3
plt.plot(x,y,'r')
plt.show()
  • 1
  • 2
  • 3
  • 4
  • 5

请添加图片描述

如果您要绘制特殊类型的线条,并想添加一些颜色,PyLab 提供了如下方法:

符号 ‘-’,’–’,’-.’,’:’,’.’,’,’,,o,^,v<,>,s,+,x,D,d,1,2,3,4,h,H,p,| ,_
颜色 b(蓝色),g(绿色),r(红色),c(青色),m(品红),y(黄色),k(黑色),w(白色)

效果如下:

from pylab import *
x = linspace(-3, 3, 30)
y = x**2
plot(x, y, 'r.')
show()
  • 1
  • 2
  • 3
  • 4
  • 5
'
运行

请添加图片描述

如果您想在同一绘图区域内绘制多个图形,只需要使用多个绘图命令。示例如下:

from matplotlib import pyplot as plt
import numpy as np
x=np.arange(-4,0,0.1)
plt.plot(x,sin(x),'r')
plt.plot(x,cos(x),'g--')
plt.plot(x,-sin(x),'b.')
plt.show()
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7

请添加图片描述

如果想清除图像,直接在程序末尾调用 clf() 方法即可。