复变函数画图cosz图像_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.