使用 python 实现 Voronoi

发布时间:2019-08-30 08:40:35编辑:auto阅读(2077)

    以下程序 使用python实现 Voronoi图
    from PIL import Image
    import random
    import math
     
    def generate_voronoi_diagram(width, height, num_cells):
    	image = Image.new("RGB", (width, height))
    	putpixel = image.putpixel
    	imgx, imgy = image.size
    	nx = []
    	ny = []
    	nr = []
    	ng = []
    	nb = []
    	for i in range(num_cells):
    		nx.append(random.randrange(imgx))
    		ny.append(random.randrange(imgy))
    		nr.append(random.randrange(256))
    		ng.append(random.randrange(256))
    		nb.append(random.randrange(256))
    	for y in range(imgy):
    		for x in range(imgx):
    			dmin = math.hypot(imgx-1, imgy-1)
    			j = -1
    			for i in range(num_cells):
    				d = math.hypot(nx[i]-x, ny[i]-y)
    				if d < dmin:
    					dmin = d
    					j = i
    			putpixel((x, y), (nr[j], ng[j], nb[j]))
    	image.save("VoronoiDiagram.png", "PNG")
            image.show()
     
    generate_voronoi_diagram(500, 500, 25)




关键字