Случайные неперекрывающиеся круги (с контролируемым номером круга) в python и pygame

Question

Я писал код для неперекрывающихся случайных кругов с разными радиусами. Я вроде получил заслуженное, но мой оператор if, который проверяет перекрытие или неперекрытие, исключает несколько кругов. Итак, я получаю меньшее количество кругов. Вот код:

import pygame
import numpy as np

pygame.init()
display_width = 800
display_height = 500

black = [0, 0, 0]
white = [255, 255, 255]
red = [255, 0, 0]

display_surface = pygame.display.set_mode((display_width, display_height))
clock = pygame.time.Clock()
pygame.display.set_caption("Random Circle")


def circle(x, y, r):
    pygame.draw.circle(display_surface, red, (int(x), int(y)), int(r), 2)


def distance(x1, y1, x2, y2):
    dsq = (x1 - x2) ** 2 + (y1 - y2) ** 2
    d = np.sqrt(dsq)
    return d


n = 100

r = np.random.randint(10, 20, size=n)
x = np.random.randint(r, display_width - r, size=n)
y = np.random.randint(r, display_height - r, size=n)

display_surface.fill(black)

for i in range(len(r)):
    valid = True
    for j in range(len(r)):
        if i != j:
            d = distance(x[i], y[i], x[j], y[j])
            if d < r[i] + r[j]:
                valid = False

    if valid:

        circle(x[i], y[i], r[i])

    pygame.display.update()
    clock.tick(100)

exit = False
while not exit:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            quit()

Я видел ответ на вопрос здесь , который делает именно то, что делает этот код (за исключением того, что он использует класс кругов, который я не использовал). Я чувствую, что мне нужно использовать какое-то время l oop, которое не заканчивается, пока я не получу желаемое количество кругов. Но мне сложно написать код. Кто-нибудь может помочь?

Answer 1

Создание списка кругов Каждый круг представляет собой кортеж с 3 компонентами, координатами x и y и радиусом. Добавляйте новые круги, если длина списка меньше n:

circle_list = []
while len(circle_list) < n:
    # [...]

Создайте случайное положение и радиус:

r = random.randint(10, 20)
x = random.randint(r, display_width - r)
y = random.randint(r, display_height - r)

Оцените, пересекается ли круг с другим кругом который находится в списке:

collide = False
for x2, y2, r2 in circle_list:
    d = distance(x, y, x2, y2)
    if d < r + r2:
        collide = True
        break

Добавить круг, если он не сталкивается:

if not collide:
    circle_list.append((x, y, r))

Пример кода:

import random

circle_list = []
while len(circle_list) < n:
    r = random.randint(10, 20)
    x = random.randint(r, display_width - r)
    y = random.randint(r, display_height - r)

    collide = False
    for x2, y2, r2 in circle_list:
        d = distance(x, y, x2, y2)
        if d < r + r2:
            collide = True
            break

    if not collide:
        circle_list.append((x, y, r))

        circle(x, y, r)
        pygame.display.update()
        clock.tick(100)