在一個需要我使用相同代碼的項目上工作,注意在同一個文件中生成mandelbrot集和julia集,我有一個工作mandelbrot集,但可以看到如何擴展到使用相同代碼的julia集。 也許沒有得到區別?任何人都可以闡述擴展mandelbrot生成julia
import numpy as np
import matplotlib.pyplot as plt
import math
def Mandelbrot(zmin, zmax, m, n, tmax=256):
xs = np.linspace(zmin, zmax, n)
ys = np.linspace(zmin, zmax, m)
X, Y = np.meshgrid(xs, ys)
Z = X + 1j * Y
C = np.copy(Z)
M = np.ones(Z.shape) * tmax
for t in xrange(tmax):
mask = np.abs(Z) <= 2.
Z[ mask] = Z[mask]**2 + C[mask]
M[-mask] -= 1.
return M
list=Mandelbrot(-2,2,500,500)
plt.imshow(list.T, extent=[-2, 1, -1.5, 1.5])
plt.gray()
plt.savefig('mandelbrot.png')
upvoting它,因爲建設性downvote – karatedog