Decide to assign a laboratory with Python (fiction)

• This case study is fiction. I have never used it for actual laboratory assignment.

Let's organize the situation

Laboratory name

#Laboratory name
labs = list("ABCDEFG")
labs

['A', 'B', 'C', 'D', 'E', 'F', 'G']

Minimum number of assignments for each laboratory

#Minimum number of assignments for each laboratory
min_assign = [2 for x in range(len(labs))]
min_assign

[2, 2, 2, 2, 2, 2, 2]

Maximum number of assignments in each laboratory

#Maximum number of assignments in each laboratory
max_assign = [4 for x in range(len(labs))]
max_assign

[4, 4, 4, 4, 4, 4, 4]

Student ID

#Student ID
n_students = 20
students = [x + 1 for x in range(n_students)]
students

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20]

Student grade ranking

#Student grade ranking
import random
grades = [x + 1 for x in range(n_students)]
random.shuffle(grades)
grades

[9, 4, 13, 14, 7, 17, 5, 2, 10, 1, 19, 16, 18, 20, 3, 15, 12, 6, 8, 11]

Laboratory assignment order survey

#Laboratory assignment order survey
questionnaire = []
for x in range(n_students):
    hope = [x + 1 for x in range(len(labs))]
    random.shuffle(hope)
    if x > 0 and random.random() < 0.1: #Make people who have the same desired order
        hope = questionnaire[x - 1]
    questionnaire.append(hope)
questionnaire

[[2, 7, 3, 5, 4, 6, 1],
 [2, 6, 1, 7, 3, 4, 5],
 [4, 7, 6, 5, 2, 1, 3],
 [6, 3, 2, 4, 1, 7, 5],
 [2, 5, 4, 3, 7, 1, 6],
 [3, 7, 2, 6, 5, 1, 4],
 [1, 7, 3, 2, 6, 5, 4],
 [4, 1, 7, 3, 6, 2, 5],
 [7, 6, 4, 3, 1, 2, 5],
 [5, 3, 4, 7, 1, 6, 2],
 [3, 2, 7, 4, 6, 1, 5],
 [6, 7, 5, 4, 1, 2, 3],
 [6, 7, 5, 4, 1, 2, 3],
 [2, 3, 7, 5, 1, 4, 6],
 [7, 3, 5, 1, 6, 4, 2],
 [6, 5, 7, 4, 3, 2, 1],
 [2, 3, 1, 5, 6, 4, 7],
 [4, 2, 1, 6, 7, 5, 3],
 [6, 2, 7, 4, 1, 3, 5],
 [2, 5, 4, 6, 1, 3, 7]]

Randomly decide the assignment

#Randomly decide the assignment
assignment = [[] for x in range(len(labs))]
rand_stu = [x for x in range(n_students)]
random.shuffle(rand_stu)
lab = 0
for stu in rand_stu:
    assignment[lab % len(labs)].append(stu)
    lab += 1

assignment

[[8, 13, 6],
 [1, 5, 17],
 [18, 16, 3],
 [9, 2, 11],
 [12, 10, 14],
 [0, 4, 15],
 [19, 7]]

Who wants what as a result of assignment

def resulted(assignment, questionnaire):
    return [[questionnaire[stu][i] for j, stu in enumerate(lab)] for i, lab in enumerate(assignment)]

resulted(assignment, questionnaire)

[[7, 2, 1], [6, 7, 2], [7, 1, 2], [7, 5, 4], [1, 6, 6], [6, 1, 2], [7, 5]]

The size of dissatisfaction as a whole

def fuman(assignment, questionnaire):
    ary = resulted(assignment, questionnaire)
    return sum([sum(x) - len(x) for x in ary])

fuman(assignment, questionnaire)

65

Prioritize those with good grades (do not consider the minimum number of assignments)

#How to prioritize the desired laboratories from those with good grades (without considering the minimum number of assignments)
assignment = [[] for x in range(len(labs))]
for i in range(len(students)):
    stu = grades.index(i + 1) #Student with grade i
    que = questionnaire[stu] #Grade ranking i-th student's desired ranking
    for j in range(len(labs)):
        lab = que.index(j + 1) #Laboratory of desired order j
        if len(assignment[lab]) < max_assign[lab]:
            assignment[lab].append(stu)
            break

assignment

[[6, 13],
 [7, 10],
 [1, 17, 16, 3],
 [14],
 [9, 18, 8, 19],
 [4, 2, 11, 5],
 [0, 15, 12]]

Who wants what as a result of assignment

resulted(assignment, questionnaire)

[[1, 2], [1, 2], [1, 1, 1, 2], [1], [1, 1, 1, 1], [1, 1, 2, 1], [1, 1, 3]]

The size of dissatisfaction as a whole

fuman(assignment, questionnaire)

6

Priority is given to those with good grades (consider the minimum number of assignments)

First fill in the minimum number of assignments

#How to prioritize the desired laboratories from those with good grades (fill in the minimum number of assignments first)
assignment = [[] for x in range(len(labs))]
assigned_students = []
for i in range(len(students)):
    stu = grades.index(i + 1) #Student with grade i
    que = questionnaire[stu] #Grade ranking i-th student's desired ranking
    for j in range(len(labs)):
        lab = que.index(j + 1) #Laboratory of desired order j
        if len(assignment[lab]) < min_assign[lab]:
            assignment[lab].append(stu)
            assigned_students.append(stu)
            break

assignment

[[6, 19], [7, 16], [1, 17], [14, 3], [9, 18], [4, 8], [0, 2]]

assigned_students

[9, 7, 14, 1, 6, 17, 4, 18, 0, 8, 19, 16, 2, 3]

Assign unassigned students

#How to prioritize the desired laboratories from those with good grades (assign unassigned students)
for i in range(len(students)):
    stu = grades.index(i + 1) #Student with grade i
    que = questionnaire[stu] #Grade ranking i-th student's desired ranking
    if stu in assigned_students:
        continue
    for j in range(len(labs)):
        lab = que.index(j + 1) #Laboratory of desired order j
        if len(assignment[lab]) < max_assign[lab]:
            assignment[lab].append(stu)
            assigned_students.append(stu)
            break

assignment

[[6, 19, 13],
 [7, 16],
 [1, 17],
 [14, 3],
 [9, 18, 11, 12],
 [4, 8, 5, 10],
 [0, 2, 15]]

Who wants what as a result of assignment

resulted(assignment, questionnaire)

[[1, 2, 2], [1, 3], [1, 1], [1, 4], [1, 1, 1, 1], [1, 2, 1, 1], [1, 3, 1]]

The size of dissatisfaction as a whole

fuman(assignment, questionnaire)

10

As a matter of fact

It's difficult because there are various laboratories where popularity is concentrated and laboratories that are not popular at all (´ ・ ω ・ `)