2016-07-29 81 views
2

我已经设法得到了一个很好的巴恩斯利蕨类植物,用下面的代码生成,但有些不太正确。 '干'有额外的线,我一直没有找到问题。有谁能解决这个问题吗 巴恩斯利蕨尽力而为:巴恩斯利蕨Python脚本

enter image description here

from PIL import Image 
import random 
import matplotlib.pyplot as plt 
A=[] 
mat=[[0.0,0.0,0.0,0.6,0.0,0.0,0.01], 
    [0.85,0.04,-0.04,0.85,0.0,1.6,0.85], 
    [0.2,-0.26,0.23,0.22,0.0,1.6,0.07], 
    [-0.15,0.28,0.26,0.24,0.0,0.44,0.07]] 
x=0.0 
y=0.0 
for k in range(0,100000): 
    p=random.random() 
    if p <= mat[0][6]: 
     i=0 
    elif p <= mat[0][6] + mat[1][6]: 
     i=1 
    elif p <= mat[0][6] + mat[1][6] + mat[2][6]: 
     i=2 
    else: 
     i=3 

    x0 = x * mat[i][0] + y * mat[i][1] + mat[i][4] 
    y = x * mat[i][2] + y * mat[i][3] + mat[i][5] 
    x = x0 

    ptn=[x,y] 

    A.append(ptn) 

plt.figure(figsize=(20,30)) 
plt.scatter(*zip(*A),marker='o', color='g',s=0.1) 
plt.show() 
+1

我不熟悉的这个分形背后的算法,但一个好的调试起点可能是为你的if语句中的不同情况分配不同的颜色。 – fodma1

回答

1

有矩阵中的错字:

mat=[[0.0,0.0,0.0,0.16,0.0,0.0,0.01], 
[0.85,0.04,-0.04,0.85,0.0,1.6,0.85], 
[0.2,-0.26,0.23,0.22,0.0,1.6,0.07], 
[-0.15,0.28,0.26,0.24,0.0,0.44,0.07]] 

这种修复它哎呀...