使用 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(50050025)  
使用 python 实现 Voronoi 图_第1张图片

你可能感兴趣的:(Python库,基础算法)