0%

【python与高等数学专题 一】Python3在一幅图绘制多个函数图像

(一)背景与应用

在高等数学等学科的学习与应用、数学分析与可视化等场景,我们时常会需要去将一个函数图像绘制出来以便于我们进行分析,通过Python3可以很方便地实现这一点,比起MATLAB等,其轻便、容易上手。

(二)用到的库

  • matplotlib
  • numpy

(三)通用代码

需求分析

我们需要一个能够在同一张图同时绘制多个函数图像的功能,以便于同时显示、对比和分析规律,因此我们先创建画布,随后根据采用plot函数进行绘制,根据函数的个数来实现多个函数的同时绘制在同一张画布上,设计的代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#!usr/bin/env python
# -*- conding:utf-8 -*-
import matplotlib.pyplot as plt
from numpy import linspace,sqrt,sin,cos,tan,arctan,pi,exp


def plotFunction(x,y,x2 = None,y2 = None,x3=None,y3 = None):
fig = plt.figure(num = 1,dpi = 120)
ax = plt.subplot(111)

# 坐标轴
ax = plt.gca() # get current axis 获得坐标轴对象
ax.spines['right'].set_color('none') # 将右边 边沿线颜色设置为空 其实就相当于抹掉这条边
ax.spines['top'].set_color('none')
ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')
# 设置中心的为(0,0)的坐标轴
ax.spines['bottom'].set_position(('data', 0)) # 指定 data 设置的bottom(也就是指定的x轴)绑定到y轴的0这个点上
ax.spines['left'].set_position(('data', 0))


ax.plot(x,y,label = "first Fig",color ="blueviolet")
if x2 is not None and y2 is not None:
ax.plot(x2,y2,label = "Second Fig",color ="red")
#ax.set_xlim(0,2)
ax.set_xlabel("line",color='cyan')
#plt.draw()
plt.legend()
plt.show()

相关的参数、图像的调整可以参考Matplotlib手册,本文旨在于提供一个方法、思想。

(四)高等数学函数测试

以函数f(x)=sin(1/x)为例,首先定义一个函数以及函数定义域范围:

1
2
3
4
x = linspace(-2*pi,2*pi,1000)
y = sin(1/x)

plotFunction(x,y)


对于复杂的三角函数,如下图所示:

如果我们想知道这个函数的图像以及与上面的sin(1/x)的差异,我们只需要定义如下:

1
2
3
4
5
x = linspace(-2*pi,2*pi,1000)
y = sin(1/x)
y2 = sqrt(x)*(x**3-4*cos(x)-sin(1))

plotFunction(x,y,x,y2)


此函数还可以进行进一步的扩展,可以通过两个函数绘图实现原函数的绘制、一阶导函数等的绘制等功能。

1
2
3
4
5
x = linspace(-pi/2,pi/2,1000)
y = arcsin(x)
y2 = 1/sqrt(1-x*x)

plotFunction(x,y,x,y2)


其他功能可自行探索/改进。

-------------结束啦 我可是有底线的~ -------------

欢迎关注我的其它发布渠道