Madra_smieciarka/QOLfunc.py
2024-03-19 21:08:24 +01:00

18 lines
485 B
Python

# Tutaj wrzucam jakieś funkcje różne, co by nie zaśmiecać tym maina jak nie trzeba
import random
def chooseNeighbours(count: int, i: int) -> set:
"""
:param count: ile sąsiadów ma wybrać
:param i: id obiektu, żeby nie był sąsiadem sam ze sobą
:return: zbiór id wybranych losowo sąsiadów obiektu
"""
neigh = set()
while len(neigh) < count:
num = random.randint(0, 7)
if num != i:
neigh.add(num)
return neigh