PySierpinskiTriangle/SierpinskiTriangle.py at main · TrainedPro/PySierpinskiTriangle · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import random
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

TriangleVertices = [(0.0, 0.0), (0.5, 1.0), (1.0, 0.0)]

x_list = []
y_list = []

figure, ax = plt.subplots()
ax.set_xlim = (0.0 , 1.0)
ax.set_ylim = (0.0 , 1.0)


line, = ax.plot(0.0, 0.0, 'g.')

def SierTri(n):
    x, y = random.choice(TriangleVertices)
    for _ in range(n):
        nx, ny = random.choice(TriangleVertices)

        x = (nx + x) / 2.0
        x_list.append(x)

        y = (ny + y) / 2.0
        y_list.append(y)

        line.set_xdata(x_list)
        line.set_ydata(y_list)

    return line,

animate = FuncAnimation(fig = figure,
                    func = SierTri,
                    frames=10,
                    interval=1)

# animate.save(r'animation.gif', fps=10)

plt.show()

print('Complete')