使用 python 实现 Voronoi 图

1. from PIL import Image  
2. import random  
3. import math  
4.    
5. def generate_voronoi_diagram(width, height, num_cells):  
6.     image = Image.new("RGB", (width, height))  
7.     putpixel = image.putpixel  
8.     imgx, imgy = image.size  
9.     nx = []  
10.     ny = []  
11.     nr = []  
12.     ng = []  
13.     nb = []  
14.     for i in range(num_cells):  
15.         nx.append(random.randrange(imgx))  
16.         ny.append(random.randrange(imgy))  
17.         nr.append(random.randrange(256))  
18.         ng.append(random.randrange(256))  
19.         nb.append(random.randrange(256))  
20.     for y in range(imgy):  
21.         for x in range(imgx):  
22.             dmin = math.hypot(imgx-1, imgy-1)  
23.             j = -1  
24.             for i in range(num_cells):  
25.                 d = math.hypot(nx[i]-x, ny[i]-y)  
26.                 if d < dmin:  
27.                     dmin = d  
28.                     j = i  
29.             putpixel((x, y), (nr[j], ng[j], nb[j]))  
30.     image.save("VoronoiDiagram.png", "PNG")  
31.         image.show()  
32.    
33. generate_voronoi_diagram(500, 500, 25)