Python 给老师分配办公室

原本是将三个老师分配到三个办公室，我稍微修改了一下：将五名学生分配给三位老师，且每位老师最多只能带两位学生。

import random
#5个学生随机分配三个老师，且每个老师带的学生数不能超过二
teachers = [[],[],[]]
students = ['a','b','c','d','e']
for student in students:
    number = random.randint(0,2)
    teacher_len1 = len(teachers[number])
    if teacher_len1 <= 1:   # 判断老师所带的学生是否超过两位
        teachers[number].append(student)
    else:
        students.append(student)    #将超过的第三位同学插入students列表重新循环
i = 0
while i <= 2:
    teacher_len2 = len(teachers[i])
    print("%d老师手下有%d名学生"%(i,teacher_len2))
    i = i+1
print(teachers)

看着老师的视频跟着写的。俺第一次发布文章，有错误的地方大家可以在评论区指出来。三克油