python/Matplotlib绘制复变函数图像教程


Posted in Python onNovember 21, 2019

今天发现sympy依赖的库mpmath里也有很多数学函数,其中也有在复平面绘制二维图的函数cplot,具体例子如下

from mpmath import *

def f1(z):
 return z

def f2(z):
 return z**3

def f3(z):
 return (z**4-1)**(1/4)

def f4(z):
 return 1/z

def f5(z):
 return atan(z)

def f6(z):
 return sqrt(z)

cplot(f1)
cplot(f2)
cplot(f3)
cplot(f4)
cplot(f5)
cplot(f6)

python/Matplotlib绘制复变函数图像教程

参照matlab绘制复变函数的例子,使用python实现绘制复变函数图像,网上还没搜到相关的文章,在这里分享出来供大家学习。

'''
参照matlab绘制复变函数的例子,创建函数cplxgrid,cplxmap,cplxroot
'''
# 1.导入相关库
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import *

# 2.创建函数
def cplxgrid(m):
 '''Return polar coordinate complex grid.

 Parameters
 ----------
 m: int

 Returns
 ----------
 z: ndarray,with shape (m+1)-by-(2*(m+1))
 '''
 m = m
 r = np.arange(0,m).reshape(m,1) / m
 theta = np.pi * np.arange(-m,m) / m
 z = r * np.exp(1j * theta)

 return z

def cplxroot(n=3,m=20):
 '''
 cplxroot(n): renders the Riemann surface for the n-th root
 cplxroot(): renders the Riemann surface for the cube root.
 cplxroot(n,m): uses an m-by-m grid. Default m = 20.

 Use polar coordinates, (r,theta).
 Use polar coordinates, (r,theta).

 Parameters
 ----------
 n: n-th root
 m: int

 Returns
 ----------
 None: Plot the Riemann surface
 '''
 m = m+1
 r = np.arange(0,m).reshape(m,1) / m
 theta = np.pi * np.arange(-n * m, n * m) / m
 z = r * np.exp(1j * theta)
 s = r * (1/n) * np.exp(1j * theta / n)
 fig = plt.figure()
 ax = fig.add_subplot(111,projection='3d')
 # ax.plot_surface(np.real(z),np.imag(z),np.real(s),color = np.imag(s))
 ax.plot_surface(np.real(z),np.imag(z),np.real(s),cmap = plt.cm.hsv)
 ax.set_xlim((-1,1))
 ax.set_ylim((-1,1))
 ax.set_xlabel('Real')
 ax.set_ylabel('Imag')
 ax.set_xticks([])
 ax.set_yticks([])
 ax.set_zticks([])
 ax.set_autoscalez_on(True)#z轴自动缩放 
 ax.grid('on')
 plt.show()

def cplxmap(z,cfun):
 '''
 Plot a function of a complex variable.

 Parameters
 ----------
 z: complex plane
 cfun: complex function to plot

 Returns
 ----------
 None: Plot the surface of complex function
 '''
 blue = 0.2
 x = np.real(z)
 y = np.imag(z)
 u = np.real(cfun)
 v = np.imag(cfun)
 M = np.max(np.max(u))#复变函数实部最大值
 m = np.min(np.min(u))#复变函数实部最大值
 s = np.ones(z.shape)
 fig = plt.figure()
 ax = fig.add_subplot(111,projection='3d')
 # 投影部分用线框图
 surf1 = ax.plot_wireframe(x,y,m*s,cmap=plt.cm.hsv)
 surf2 = ax.plot_surface(x,y,u,cmap=plt.cm.hsv)

 #绘制复变函数1/z时会出错,ValueError: Axis limits cannot be NaN or Inf
 # ax.set_zlim(m, M) 
 ax.set_xlim((-1,1))
 ax.set_ylim((-1,1))
 ax.set_xlabel('Real')
 ax.set_ylabel('Imag')
 ax.set_xticks([])
 ax.set_yticks([])
 ax.set_zticks([])
 ax.set_autoscalez_on(True)#z轴自动缩放

 ax.grid('on')
 plt.show()

def _test_cplxmap():
 '''测试cplxmap函数'''
 z = cplxgrid(30)
 w1 = z
 w2 = z**3
 w3 = (z**4-1)**(1/4)
 w4 = 1/z
 w5 = np.arctan(2*z)
 w6 = np.sqrt(z)
 w = [w1,w2,w3,w4,w5,w6]
 for i in w:
 cplxmap(z,i)

def _test_cplxroot():
 '''测试cplxroot函数'''
 cplxroot(n=2)
 cplxroot(n=3)
 cplxroot(n=4)
 cplxroot(n=5)

if __name__ == '__main__':
 _test_cplxmap()
 _test_cplxroot()

python/Matplotlib绘制复变函数图像教程

python/Matplotlib绘制复变函数图像教程

以上这篇python/Matplotlib绘制复变函数图像教程就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python cookielib 登录人人网的实现代码
Dec 19 Python
在Windows8上的搭建Python和Django环境
Jul 03 Python
Python实现对比不同字体中的同一字符的显示效果
Apr 23 Python
基于python3 类的属性、方法、封装、继承实例讲解
Sep 19 Python
Python 3.3实现计算两个日期间隔秒数/天数的方法示例
Jan 07 Python
python requests库爬取豆瓣电视剧数据并保存到本地详解
Aug 10 Python
python 字典访问的三种方法小结
Dec 05 Python
selenium+python配置chrome浏览器的选项的实现
Mar 18 Python
Django import export实现数据库导入导出方式
Apr 03 Python
python爬取股票最新数据并用excel绘制树状图的示例
Mar 01 Python
Python+Selenium实现读取网易邮箱验证码
Mar 13 Python
Python必备技巧之字符数据操作详解
Mar 23 Python
Python 复平面绘图实例
Nov 21 #Python
Python3并发写文件与Python对比
Nov 20 #Python
python实现高斯(Gauss)迭代法的例子
Nov 20 #Python
python、Matlab求定积分的实现
Nov 20 #Python
python 求定积分和不定积分示例
Nov 20 #Python
python 计算积分图和haar特征的实例代码
Nov 20 #Python
Python 函数绘图及函数图像微分与积分
Nov 20 #Python
You might like
Laravel框架使用Redis的方法详解
2018/05/30 PHP
prototype1.4中文手册
2006/09/22 Javascript
JS类库Bindows1.3中的内存释放方式分析
2007/03/08 Javascript
JavaScript 继承机制的实现(待续)
2010/05/18 Javascript
基于jquery的无刷新分页技术
2011/06/11 Javascript
js监听输入框值的即时变化onpropertychange、oninput
2011/07/13 Javascript
javascript 获取模态窗口的滚动位置代码
2013/08/06 Javascript
关于Javascript作用域链的八点总结
2013/12/06 Javascript
javascript轻量级库createjs使用Easel实现拖拽效果
2016/02/19 Javascript
学习Angularjs分页指令
2016/07/01 Javascript
javascript 数组的正态分布排序的问题
2016/07/31 Javascript
JS之获取样式的简单实现方法(推荐)
2016/09/13 Javascript
老生常谈javascript变量的命名规范和注释
2016/09/29 Javascript
Bootstrap优化站点资源、响应式图片、传送带使用详解3
2016/10/14 Javascript
Vue.js常用指令汇总(v-if、v-for等)
2016/11/03 Javascript
node.js请求HTTPS报错:UNABLE_TO_VERIFY_LEAF_SIGNATURE\的解决方法
2016/12/18 Javascript
详解如何将angular-ui的图片轮播组件封装成一个指令
2017/05/09 Javascript
JS排序算法之希尔排序与快速排序实现方法
2017/12/12 Javascript
NodeJS爬虫实例之糗事百科
2017/12/14 NodeJs
关于Vue在ie10下空白页的debug小结
2018/05/02 Javascript
VueJS 组件参数名命名与组件属性转化问题
2018/12/03 Javascript
JS计算两个数组的交集、差集、并集、补集(多种实现方式)
2019/05/21 Javascript
Linux 修改Python命令的方法示例
2018/12/03 Python
Python类装饰器实现方法详解
2018/12/21 Python
Python 计算任意两向量之间的夹角方法
2019/07/05 Python
Django中间件基础用法详解
2019/07/18 Python
基于python cut和qcut的用法及区别详解
2019/11/22 Python
英国护肤品购物网站:Beauty Expert
2016/08/19 全球购物
潘多拉珠宝英国官方网上商店:PANDORA英国
2018/06/12 全球购物
澳大利亚家庭花园和DIY工具网店:VidaXL
2019/05/03 全球购物
学院领导推荐信
2013/10/30 职场文书
DIY蛋糕店的创业计划书范文
2013/12/26 职场文书
超市优秀员工事迹材料
2014/05/01 职场文书
2014县委书记党的群众路线教育实践活动对照检查材料思想汇报
2014/09/22 职场文书
基层工作经验证明样本
2014/11/16 职场文书
MySQL 如何设计统计数据表
2021/06/15 MySQL