python/Matplotlib绘制复变函数图像教程
作者:落叶_小唱 发布时间:2023-08-03 07:36:43
标签:python,Matplotlib,复变函数
今天发现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)
参照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()
来源:https://blog.csdn.net/ouening/article/details/82349372
0
投稿
猜你喜欢
- 导言到目前为止的讨论编辑DataList的教程里,没有包含任何验证用户的输入,即使是用户非法输入— 遗漏了product的name或者负的p
- class和id的命名,如果合理,可以使得文档具有清晰的结构我们现在解决办法就是使用现有的元素,通过给他们id或class而得到额外的信息。
- 在编程时你一定碰到过时间触发的事件,在VB中有timer控件,而asp中没有, 假如你要不停地查询数据库来等待一个返回结果的话,我想你一定知
- 首先为什么会有axis这个概念?因为在numpy模块中,大多数处理的是矩阵或者多维数组,同时,对多维数组或者矩阵的操作有多种可能,为了帮助实
- 有些使用Z-Blog的用户询问我,怎么实现我的月光博客首页上这种自动图文混排的版式效果,今天我就详细介绍一下在Z-Blog中实现这种图文混排
- [本站原创]在我们浏览了一些网页时,经常会弹出一些信息窗口或浏览器窗口以显示一些公告内容,想知道这些窗口是怎么制作出来的吗?如果你还不曾知道
- 写好脚本,注册好服务之后,经测试,ORACLE可以随RHEL启动而启动,但不能随系统关闭而关闭。在网上找答案,发现几乎所有的设置过程帖子都是
- 细线边框是网页中定位区分内容常用的一种方法,配合特定图片的使用,往往能够达到不错的效果,那么如何制作细线边框呢?asp之家注:现在要实现这个
- <% dim week_ymd(8) '测出可以手动设定日期,比如this_ymd=#2008-04-1
- 相信每个前端工程师都有自己喜爱的javascript框架,说情感也好,道信仰也罢,javascript框架带给人的不仅仅是便捷的开发,更有一
- 任何一位数据库程序员都会有这样的体会:高通信量的数据库驱动程序中,一条糟糕的SQL查询语句可对整个应用程序的运行产生严重的影响,其不仅消耗掉
- 本文实例讲述了Golang最大递减数算法问题。分享给大家供大家参考,具体如下:给出一个非负整数,找到这个非负整数中包含的最大递减数。一个数字
- 适用环境: PHP5.2.x / mysql 5.0.xclass Mysql { priva
- 这里使用FSO对象来删除指定文件夹 代码和说明如下:<% set fs=createobject(&qu
- 平时在PL/SQL中的编程中遇到一些问题,这里以问答的形式来进行把它们总结下来,以供大家分享。1、当需要向表中装载大量的数据流或者需要处理大
- 经常到这来抄抄改改代码,也贡献一个代码,主要是讨论研究用,没有封装可以进一步改进<!DOCTYPE html PUBLIC "
- 数据列类型与查询效率选用适当的数据列类型有助于提高查询命令的执行速度,下面是几点关于如何选择合适数据列类型的建议:尽量选用尺寸较小的数据列。
- clipboardData 对象提供了对于预定义的剪贴板格式的访问,以便在编辑操作中使用。成员表方法 描述 clearData 通过 dat
- 举例: 如:在字段名处输入:username,password,email,telphone 注意:不同的字段名用英文逗号隔开,且不支持星号
- Oracle分页存储过程的思路于sqlserver的思路是一样的,但是我这里做了点改动,在因为Oracle的语法和规则的不同所以,Oracl