python编程matplotlib交互绘制Julia集示例解析
作者:微小冷 发布时间:2021-07-31 17:14:45
标签:matplotlib,Julia,交互绘制
所谓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集。
在mpl中,事件绑定函数mpl_connect被封装在cavnas类中,调用格式为
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集的分形特征意味着我们所生成的图片可以无限放大,但是mpl自带的放大工具并不会重新生成数据,所以是虚假的放大。因此需要重新绑定放大操作,其思路是,当右键点击(‘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)
来源:https://blog.csdn.net/m0_37816922/article/details/102530064


猜你喜欢
- 在pandas 基础操作大全之数据读取&清洗&分析中介绍了pandas常见的数据处理操作,现在继续对pandas常用的数据合
- LEN()函数获取字符串的长度。LEN( <字符> )--返回整型SUBSTRING()函数截取字符内指定位置、指定内容的字符。
- Python中的垃圾回收算法是采用引用计数, 当一个对象的引用计数为0时, Python的垃圾回收机制就会将对象回收a = "la
- Js代码:/*** 验证码*/function yzm(){var codeChars = new Array(0, 1, 2, 3, 4,
- 表的普通字段 一对多字段 多对多字段 插入数据#插入数据def add(request):G_title=request.POST.get(
- 目前,SQL Server数据库有几个版本都在使用中,比如 7.0, 2000和2005,那么,在现实的工作和学习中,你很有可能会需要从以前
- 1.使用SQL*PLUS停止数据库$ sqlplus /nologSQL> connect / as sysdbaSQL> sh
- python之pkl文件pkl文件是python里面保存文件的一种格式,如果直接打开会显示一堆序列化的东西。cPickle在python3中
- 年初的时候收藏过一篇关于mysqlreport的报表解读,和内置的show status,和show variables相比mysqlrep
- 对于yolo系列,应用广泛,在win10端也有很大的应用需求,所以这篇文章给出win10环境下的安装教程。先给出系列文章win10+anac
- 在日常开发中,我们进行用户登录的时候,大部分情况下都会使用 session 来保存用户登录信息,并以此为依据判断用户是否已登录。但其实 HT
- 先备份数据库,再用下面的办法: USE MASTER GO SP_CONFIGURE 'ALLOW UPDATES',1 R
- 本文实例讲述了Python实现对excel文件列表值进行统计的方法。分享给大家供大家参考。具体如下:#!/usr/bin/env pytho
- type()函数:使用type()函数可以判断对象的类型,如果一个变量指向了函数或类,也可以用type判断。如:class Student(
- 上文中,我们一起学习了手把手教你实现一个 Python 计时器。本文中,云朵君将和大家一起了解什么是上下文管理器 和 Python
- 优化算法时间复杂度算法的时间复杂度对程序的执行效率影响最大,在Python中可以通过选择合适的数据结构来优化时间复杂度,如list和set查
- 一、图示上面为pdf截图内容,下面为转化后的word截图内容接下来,我们试试自己动作写这个工具吧!二、前期准备由于我们采用的是python进
- 有时候我们可能不知道一个用户的密码,但是又需要以这个用户做一些操作,又不能去修改掉这个用户的密码,这个时候,就可以利用一些小窍门,来完成操作
- Python关于mySQL的连接插件众多,Bottle下也有人专门开发的插件:bottle-mysql具体使用方法见官方,总共感觉其用法限制
- 模块内部封装了很多实用的功能,有时在模块外部调用就需要将其导入。常见的方式有如下几种:1 . import>>> impo