python繪制超炫酷動(dòng)態(tài)Julia集示例

更新時(shí)間：2021年12月13日 16:49:39 作者：微小冷

大家好，本篇文章主要講的是python繪制超炫酷動(dòng)態(tài)Julia集示例，感興趣的痛學(xué)趕快來(lái)看一看吧，對(duì)你有幫助的話(huà)記得收藏一下，方便下次瀏覽

前言

此Julia非彼Julia，指的是對(duì)于某復(fù)數(shù) c c c，使得迭代式 f ( z ) = z 2 + c f(z)=z^2+c f(z)=z2+c收斂的復(fù)數(shù) z z z的集合。例如，當(dāng) c = 0 c=0 c=0時(shí)，那么其收斂區(qū)間為 z 2 < 1 z^2<1 z2<1的單位圓，對(duì)應(yīng)的 c c c的Julia集便是 cos ? θ + i sin ? θ \cos\theta+i\sin\theta cosθ+isinθ。

Mandelbrot集

特別地，當(dāng) c = z c=z c=z的初始值時(shí)，符合收斂條件的 z z z的便構(gòu)成大名鼎鼎的Mandelbrot集

在這里插入圖片描述

在上圖中，顏色表示該點(diǎn)的發(fā)散速度，可以理解為開(kāi)始發(fā)散時(shí)迭代的次數(shù)。其生成代碼也非常簡(jiǎn)單，唯一需要注意的是，由于使用了大量的矩陣運(yùn)算，故使用了cupy，如果電腦沒(méi)裝cuda，只需將所有的cp改為np即可。

# 這些代碼會(huì)在后面的程序中反復(fù)調(diào)用，不再說(shuō)明
import numpy as np
import time
import matplotlib.pyplot as plt
import cupy as cp

#生成z坐標(biāo) x0,y0 為起始點(diǎn)， nx,ny為點(diǎn)數(shù)， delta為點(diǎn)距
def genZ(x0, y0, nx, ny, delta):
    real, img = cp.indices([nx,ny])*delta
    real += x0
    img += y0
    return real.T+img.T*1j

#獲取Julia集，n為迭代次數(shù)，m為判定發(fā)散點(diǎn)，大于1即可
def getJulia(z,c,n,m=2):
    t = time.time()
    z,out = z*1, cp.abs(z)
    c = cp.zeros_like(z)+c
    for i in range(n):
        absz = cp.abs(z)
        z[absz>m]=0		#對(duì)開(kāi)始發(fā)散的點(diǎn)置零
        c[absz>m]=0		
        out[absz>m]=i	#記錄發(fā)散點(diǎn)的發(fā)散速度
        z = z*z + c
    print("time:",time.time()-t)
    return out

z1 = genZ(-2,-1.5,1000,1000,0.003)
mBrot = getJulia(z1,z1,50)
plt.imshow(mBrot.get(), cmap=plt.cm.jet)
plt.show()

如果對(duì)其生成過(guò)程感興趣，那么可以觀(guān)察一下隨著迭代次數(shù)的增加，圖像的變化情況

在這里插入圖片描述

代碼如下。

from matplotlib import animation

fig = plt.figure()
fig.subplots_adjust(top=1, bottom=0, left=0, right=1)
ax = plt.subplot()

def getJulias(z,c,n,m=2):
    z,out = z*1, cp.abs(z)
    c = cp.zeros_like(z)+c
    J = []
    for i in range(n):
        z = z*z + c
        absz = cp.abs(z)
        z[absz>m]=0		#對(duì)開(kāi)始發(fā)散的點(diǎn)置零
        c[absz>m]=0		
        out[absz>m]=i	#記錄發(fā)散點(diǎn)的發(fā)散速度
        im = ax.imshow(out.get(),cmap=plt.cm.jet, animated=True)
        ax.set_axis_off()
        J.append([im])
    return J

N = 75     #迭代次數(shù)
z1 = genZ(-2,-1.5,1000,1000,0.003)
J = getJulias(z1,z1,N)

ani = animation.ArtistAnimation(fig, J, interval=50, blit=True,repeat_delay=1000)
plt.show()
ani.save('julias.gif',writer='imagemagick')

無(wú)限縮放

Mandelbrot集的分形特征意味著我們所生成的圖片可以無(wú)限放大，但是受到柵格化尺寸的影響，手動(dòng)的放大并不會(huì)更改其真實(shí)尺寸，

為了照顧觀(guān)感，將縮放中心作為圖像的中心，所以對(duì)genZ函數(shù)進(jìn)行修改。如果選取(-0.75,-0.2)作為縮放中心，則其變化如下

在這里插入圖片描述

代碼為

from matplotlib import animation

# 生成z坐標(biāo) xy=np.array([xc,yc]) 為起始點(diǎn)，
# nxy=np.array([nx,ny])為點(diǎn)數(shù)， delta為點(diǎn)距
def genZbyCenter(xy,nxy,delta):
    x0, y0 = xy-np.array(nxy)*delta/2
    return genZ(x0,y0,*nxy,delta)

mBrots = []
xy = [-0.75,-0.2]
nxy = [1000,1000]
delta0 = 0.003  #初始寬度

fig = plt.figure()
fig.subplots_adjust(top=1, bottom=0, left=0, right=1)
ax = plt.subplot()

for n in range(50):
    z1 = genZbyCenter(xy,nxy,1.1**(-n)*delta0)
    out = getJulia(z1,z1,40)
    im = ax.imshow(out.get(),cmap=plt.cm.jet, animated=True)
    ax.set_axis_off()
    mBrots.append([im])

ani = animation.ArtistAnimation(fig, mBrots, interval=50, blit=True)
plt.show()
ani.save('zoom.gif',writer='imagemagick')

Julia集

如果更改c的值，那么就能得到一個(gè)變化著的Julia集，例如，下面選取一條直線(xiàn)

y = x y=x y=x

上面的Julia集，效果如圖所示

在這里插入圖片描述

代碼為

z1 = genZ(-2,-1.5,1000,1000,0.003)

fig = plt.figure()
fig.subplots_adjust(top=1, bottom=0, left=0, right=1)
ax = plt.subplot()

mBrots = []
for x in np.arange(0.5,1,0.01):
    c = x + x*1j
    out = getJulia(z1,c,40)
    im = ax.imshow(out.get(),cmap=plt.cm.jet, animated=True)
    ax.set_axis_off()
    mBrots.append([im])

ani = animation.ArtistAnimation(fig, mBrots, interval=50)
plt.show()
ani.save('julia.gif',writer='imagemagick')

到此這篇關(guān)于python繪制超炫酷動(dòng)態(tài)Julia集示例的文章就介紹到這了,更多相關(guān)python繪制動(dòng)態(tài)Julia集內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: