Python даже групповое назначение

Проблема заключается в случайном выборе команды для каждого игрока. Поскольку random.randint производит одинаково распределенные значения, каждый игрок имеет одинаковые шансы быть назначенным для любой команды, поэтому вы можете оказаться в команде в одной команде.

Вместо этого вы должны рассмотреть возможность повторения команд и присвоение ему случайного игрока.

Плохое осуществление этой идеи будет похоже на

>>> import random
>>> 
>>> groupnum = 2
>>> peoplenum = 8
>>> 
>>> people = [i for i in range(peoplenum)]
>>> 
>>> for i in range(peoplenum):
...     group = i % groupnum
...     person = random.choice(people)
...     people.remove(person)
...     print('Number {} assigned to {}'.format(person, group))
... 
Number 6 assigned to 0
Number 7 assigned to 1
Number 4 assigned to 0
Number 3 assigned to 1
Number 2 assigned to 0
Number 5 assigned to 1
Number 1 assigned to 0
Number 0 assigned to 1

Это решит проблему, но она полагается на вызов remove чтобы избежать повторения членов команды. Чтобы этого избежать, вы можете перетасовать список игроков (делая его случайным) и zip результаты в списке команд.

>>> import random
>>>
>>> players = [i for i in range(peoplenum)]
>>> teams = [i for i in range(groupnum)]
>>> 
>>> random.shuffle(players)
>>> 
>>> [x for x in zip(players, teams)]
[(7, 0), (5, 1)]

Это явно не работает. Это связано с тем, что zip остановится, когда самый короткий итератор остановится, в наших groups. Мы хотим, чтобы он повторялся, пока есть игроки. Мы можем достичь такой функциональности с помощью itertools.cycle:

>>> import random
>>> import itertools
>>> 
>>> players = [i for i in range(peoplenum)]
>>> teams = itertools.cycle(range(groupnum))
>>> 
>>> random.shuffle(players)
>>> 
>>> [x for x in zip(players, teams)]
[(7, 0), (2, 1), (0, 0), (1, 1), (5, 0), (4, 1), (3, 0), (6, 1)]

Используя только стандартную библиотеку, я бы решил это следующим образом:

import random
from itertools import accumulate


def print_groups(n, g):
    # Prepare group separators
    size = n // g
    rem = n % g
    separators = list(accumulate([0] + [size+1] * rem + [size] * (g - rem)))

    # Make raw data
    items = list(range(n))
    random.shuffle(items)

    # Iterate and print
    for i, s in enumerate(zip(separators, separators[1:])):
        group = items[slice(*s)]
        print(f'Group {i+1}: {group} (size {len(group)})')

Если количество ваших людей делится на ваше количество групп, все группы будут иметь одинаковый размер, в противном случае первые группы n % g получат один дополнительный член.

Пример 1:

print_groups(12, 4)

Group 1: [6, 11, 10] (size 3)
Group 2: [7, 2, 5] (size 3)
Group 3: [3, 1, 9] (size 3)
Group 4: [4, 8, 0] (size 3)

Пример 2:

print_groups(14, 4)

Group 1: [8, 3, 4, 6] (size 4)
Group 2: [1, 11, 0, 12] (size 4)
Group 3: [7, 5, 9] (size 3)
Group 4: [13, 10, 2] (size 3)

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

import random
import sys


def main():
    while True:

        try:
            print("Welcome to this group picker, follow the instructions and your groups will be picked.")
            groupnum = int(input("How many groups do you want?"))
            peoplenum = int(input("How many people are there?"))
            if groupnum == 'Q':
                sys.exit(0)
            print("Okay, assign everyone a number from 1 to", peoplenum, ".")

            nums = []
            for i in range(1, peoplenum+1):
                nums.append(i)

            ''' Validate number of people vs groups '''
            if peoplenum % groupnum:
                print("error: You have incorrect number of people to divide them equally")
                sys.exit(1)

            ''' People per team '''
            peoplePerTeam = int(peoplenum/groupnum)

            ''' Initialize the group'''
            group = {}
            for i in range(1,groupnum+1):
                group[i] = set()

            ''' Loop through the people'''
            for person in nums:
                while True:
                    ''' Find random team '''
                    randomGroup = random.randint(1,groupnum)

                    ''' If team is not filled up yet, then add otherwise find a new team'''
                    if len(group[randomGroup]) < peoplePerTeam:
                        group[randomGroup].add(person)
                        break

            ''' Display the team info'''    
            for i in range(1,groupnum+1):
                print("Team %s members are %s" %(i, group[i]))

        except:
            print("Error, please follow instructions and enter only numbers.")
            break


main()

Пример выполнения этой программы приведен ниже:

Welcome to this group picker, follow the 
instructions and your groups will be picked.
How many groups do you want?2
How many people are there?8
Okay, assign everyone a number from 1 to 8 .
**Team 1 members are {1, 2, 5, 7}**
**Team 2 members are {8, 3, 4, 6}**
Welcome to this group picker, follow the 
instructions and your groups will be picked.
How many groups do you want?4
How many people are there?20
Okay, assign everyone a number from 1 to 20 .
**Team 1 members are {1, 18, 19, 20, 17}**
**Team 2 members are {16, 13, 11, 5, 7}**
**Team 3 members are {10, 9, 2, 3, 4}**
**Team 4 members are {8, 12, 15, 6, 14}**
Welcome to this group picker, follow the instructions and your groups will be picked.

Отрегулируйте вероятность каждой группы в соответствии с количеством групп. Используйте цикл while, который постоянно назначает группы и возвращается только после того, как группы имеют равную сумму.

Я бы сделал это с помощью numpy.random.choice, заменив значение false. См. Соответствующую документацию по количеству

Попробуйте сыграть с этим кодом:

import numpy as np

def assign_people_to_teams(people_count = 8, team_count = 2):
    """
        selects people_count random numbers in the range 0 to people_count-1 without replacement
        then make a team assignment decision based on desired number of teams.
    """
    return [
        element % team_count for element in list(
            np.random.choice(
                people_count,
                people_count,
                replace=False
            )
        )
    ]

team_assignment = assign_people_to_teams()

print(team_assignment)

Результат:

[0, 0, 1, 0, 0, 1, 1, 1]