Python Matplotlib交互绘制Julia集与Mandelbrot集详解
发布时间：2026-10-15 |
6人浏览过
本文详解如何使用Python Matplotlib库实现Julia集与Mandelbrot集的交互式绘制。通过绑定鼠标点击、缩放及滚轮事件，动态计算并重新生成分形图像，解决内置工具无法真实放大的问题，提供完整的代码逻辑与事件处理机制解析。
Julia集呈现出如上图所示的精美视觉效果。
Julia集
当参数c等于z的初始值时，满足收敛条件的z值集合构成了著名的Mandelbrot集。
图中颜色代表点的发散速度，即开始发散时的迭代次数。其生成代码实现如下：
#mbrot.py
import numpy as np
import time
import pyplotlib.pyplot as plt
#生成z坐标，axis为起始位置，nx，ny为x向和y向的格点个数
def genZ(axis,nx,ny):
x0,x1,y0,y1 = axis
x = np.linspace(x0,x1,nx)
y = np.linspace(y0,y1,ny)
real, img = np.meshgrid(x,y)
z = real + img*1j
return z
#获取Julia集，n为迭代次数，m为判定发散点，大于1即可
def getJulia(z,c,n,m=2):
t = time.time()
c = np.zeros_like(z)+c
out = abs(z)
for i in range(n):
absz = abs(z)
z[absz>m]=0 #对开始发散的点置零
c[absz>m]=0
out[absz>m]=i #记录发散点的发散速度
z = z*z + c
print("time:",time.time()-t)
return out
if __name__ == "__main__":
axis = np.array([-2,1,-1.5,1.5])
z0 = genZ(axis,500,500)
mBrot = getJulia(z0,z0,50)
plt.imshow(mBrot, cmap=cm.jet, extent=axis)
plt.gca().set_axis_off()
plt.show()
Matplotlib事件绑定
目标是实现点击Mandelbrot集中的某一点，从而生成对应的Julia集。在Matplotlib中，事件绑定函数mpl_connect封装在canvas类中，调用格式为：
canvas.mpl_connect('str', func)
其中func为事件处理函数，字符串参数为传入该函数的事件标识，具体列表如下，可根据名称理解其含义：
'button_press_event'
'button_release_event'
'draw_event'
'key_press_event'
'key_release_event'
'motion_notify_event'
'pick_event'
'resize_event'
'scroll_event'
'figure_enter_event'
'figure_leave_event'
'axes_enter_event'
'axes_leave_event'
'close_event'
为简化演示，首先检测鼠标点击事件'button_press_event'。需要定义一个事件处理函数，并对上述入口函数进行适当修改：
def test(evt):
print(evt.xdata) #xdata即x方向的坐标
if __name__ == "__main__":
axis = np.array([-2,1,-1.5,1.5])
z0 = genZ(axis,500,500)
mBrot = getJulia(z0,z0,50)
fig, ax = plt.subplots()
fig.canvas.mpl_connect('button_press_event', test)#调用事件函数
plt.imshow(mBrot, cmap=cm.jet, extent=axis)
plt.gca().set_axis_off()
plt.show()
执行后，点击imshow()生成的图片即可返回相应的x坐标值。
python mbrot.py
time: 0.47572827339172363
-0.8652597402597402
-0.7840909090909087
-0.18344155844155807
0.23051948051948123
0.8149350649350655
缩放功能
生成Julia集只需重新调用一次getJulia函数即可。由于Mandelbrot集具有分形特征，生成的图像可以无限放大，但Matplotlib自带的放大工具仅对现有数据进行插值放大，并非重新生成数据，因此属于虚假放大。为此，需要重新绑定放大操作：当右键点击（'button_press_event'）时记录坐标，右键释放（'button_release_event'）时重新绘制图像。为避免与左键事件冲突，需在对应的事件处理函数中加入左右键判断逻辑。
运行结果如图所示。
此外，还可以绑定鼠标滚轮事件，实现Mandelbrot集在该点的真实缩放，相关代码如下：
import matplotlib.pyplot as plt
import numpy as np
from matplotlib import cm
import matplotlib.backend_bases as mbb
import time
class MandelBrot():
def __init__(self,x0,x1,y0,y1,n):
self.oriAxis = np.array([x0,x1,y0,y1]) #初始坐标
self.axis = self.oriAxis
self.nx,self.ny,self.nMax = n,n,n #x,y方向的网格划分个数
self.nIter = 100 #迭代次数
self.n0 = 0 #预迭代次数
self.z = genZ(self.oriAxis,self.nx,self.ny)
self.DrawMandelbrot()
def DrawMandelbrot(self):
mBrot = getJulia(self.z,self.z,self.nIter)
self.fig, ax = plt.subplots()
plt.imshow(mBrot, cmap=cm.jet, extent=self.axis)
plt.gca().set_axis_off()
self.fig.canvas.mpl_disconnect(self.fig.canvas.manager.key_press_handler_id)
self.fig.canvas.mpl_connect('button_press_event', self.OnMouse)
self.fig.canvas.mpl_connect('button_release_event', self.OnRelease)
self.fig.canvas.mpl_connect('scroll_event', self.OnScroll)
plt.show()
def DrawJulia(self,c0):
z = genZ([-2,2,-2,2],800,800)
julia = getJulia(z,c0,self.nIter)
jFig,jAx = plt.subplots()
plt.cla()
plt.imshow(julia, cmap=cm.jet, extent=self.axis)
plt.gca().set_axis_off()
plt.show()
jFig.canvas.draw_idle()
#滚轮缩放
def OnScroll(self,evt):
x0,y0 = evt.xdata,evt.ydata
if evt.button == "up":
self.axis = (self.axis+[x0,x0,y0,y0])/2
elif evt.button == 'down':
self.axis = 2*self.axis-[x0,x0,y0,y0]
z = genZ(self.axis,self.nx,self.ny)
mBrot = getJulia(z,z,self.nIter)
plt.cla()
plt.imshow(mBrot, cmap=cm.jet, extent=self.axis)
plt.gca().set_axis_off()
mBrot[mBrot<1]==self.n0+self.nIter
self.n0 = int(np.min(mBrot))
self.fig.canvas.draw_idle()
pass
def OnMouse(self, evt):
self.xStart = evt.xdata
self.yStart = evt.ydata
self.fig.canvas.draw_idle()
def OnRelease(self,evt):
x0,y0,x1,y1 = self.xStart,self.yStart,evt.xdata,evt.ydata
if evt.button == mbb.MouseButton.LEFT:
self.DrawJulia(x1+y1*1j) #如果释放的是左键，那么就绘制Julia集并返回
return
#右键拖动，可以对Mandelbrot集进行真实的放大
self.axis = np.array([min(x0,x1),max(x0,x1),
min(y0,y1),max(y0,y1)])
nxny = self.axis[[1,3]]-self.axis[[0,2]]
self.nx,self.ny = (nxny/max(nxny)*self.nMax).astype(int)
z = genZ(self.axis,self.nx,self.ny)
n = 100 #n为迭代次数
mBrot = getJulia(z,z,n)
plt.cla()
plt.imshow(mBrot, cmap=cm.jet, extent=self.axis)
plt.gca().set_axis_off()
mBrot[mBrot<1]==self.n0+n
self.n0 = int(np.min(mBrot))
self.fig.canvas.draw_idle()
def genZ(axis,nx,ny):
x0,x1,y0,y1 = axis
x = np.linspace(x0,x1,nx)
y = np.linspace(y0,y1,ny)
real, img = np.meshgrid(x,y)
z = real + img*1j
return z
def getJulia(z,c,n,n0=0,m=2):
t = time.time()
c = np.zeros_like(z)+c
out = abs(z)
for _ in range(n0):
z = z*z + c
for i in range(n0,n0+n):
absz = abs(z)
z[absz>m]=0
c[absz>m]=0
out[absz>m]=i
z = z*z + c
print("time:",time.time()-t)
return out
if __name__ == "__main__":
x,y = 0,0
brot = MandelBrot(-2,1,-1.5,1.5,1000)
以上为Python编程中利用Matplotlib交互绘制Julia集的示例解析，更多关于Matplotlib交互绘制Julia集的资料，敬请关注本站其他相关文章！
本站声明
本文内容仅供参考，不代表本站观点。转载或引用请注明出处。如发现内容问题，请通过联系我们页面反馈。
