import matplotlib.pyplot as plt
import numpy as np
def koch_snowflake(order, scale=10):
? ? def _koch_snowflake_complex(order):
? ? ? ? if order == 0:
? ? ? ? ? ? # 初始三角形
? ? ? ? ? ? angles = np.array([0, 120, 240]) + 90
? ? ? ? ? ? return scale / np.sqrt(3) 雪花图案 * np.exp(np.deg2rad(angles) * 1j)
? ? ? ? else:
? ? ? ? ? ? ZR 
= 0.5 - 0.5j * np.sqrt(3) / 3
? ? ? ? ? ? p1 = _koch_snowflake_complex(order - 1) ?# 起点
? ? ? ? ? ? p2 = np.roll(p1, shift=-1) ?# 终点
? ? ? ? ? ? dp = p2 - p1 ?# 连接向量
? ? ? ? ? ? new_points = np.empty(len(p1) * 4, dtype=np.complex128)
? ? ? ? ? ? new_points[::4] = p1
? ? ? ? ? ? new_points[1::4] = p1 + dp / 3
? ? ? ? ? ? new_points[2::4] = p1 + dp * ZR
? ? ? ? ? ? new_points[3::4] = p1 + dp / 3 * 2
? ? ? ? ? ? return new_points
? ? points = _koch_snowflake_complex(order)
? ? x, y = points.real, points.imag
雪花图案 ? ? return x, y
x,y=koch_snowflake(order=2)
fig=plt.figure()
ax=fig.add_subplot(111)
ax.fill(x,y,facecolor='lightsalmon',edgecolor='orangered',linewidth=3)
plt.show() ? ? ??
雪花图案
