python вороной как сделать не грань между соседними точками - PullRequest
Новая
       46

python вороной как сделать не грань между соседними точками

0 голосов
/ 03 марта 2020

У меня есть набор точек и я хочу вычислить диаграмму Вороного. Это работает, но я хотел бы изменить его таким образом, чтобы при наличии двух точек рядом друг с другом не создавалось никаких дополнительных ребер, только если расстояние между ними превышает некоторый порог.

Кто-нибудь знает, как это сделать?

На данный момент я использую этот код: 

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)
...