フラクタル図形 － シェルピンスキーのギャスケット

import math
import matplotlib.pyplot as plt
import matplotlib.patches as patch

# 階層深さ(n)、一辺の長さ(size)、左下位置(offset)
n = 7
size = 100
offset = [0, 0]

# シェルピンスキーを計算する関数
def sierpinski(n, size, offset):
    if n == 0:
        return

    a = offset
    b = [a[0]+size/2, a[1]+size*math.sqrt(3)/2]
    c = [a[0]+size, a[1]]
    s = [a[0]+size/4, (a[1]+b[1])/2]
    t = [a[0]+size*3/4, (a[1]+b[1])/2]
    u = [a[0]+size/2, a[1]]

    if n == 1:
        tri1 = patch.Polygon([a,s,u], fill = True)
        tri2 = patch.Polygon([s,b,t], fill = True)
        tri3 = patch.Polygon([u,t,c], fill = True)
        ax.add_patch(tri1)
        ax.add_patch(tri2)
        ax.add_patch(tri3)
        
    sierpinski(n-1, size/2, a)
    sierpinski(n-1, size/2, s)
    sierpinski(n-1, size/2, u)

# Plot作成
fig = plt.figure(figsize=(5, 5))
ax = fig.add_subplot(1,1,1)
sierpinski(n, size, offset)

# Plotの設定
ax.set_xlim(offset[0], offset[0] + size)
ax.set_ylim(offset[1], offset[1] + size)
ax.set_aspect('equal')

# 描画
plt.show()