用Python怎么写哎，各位。

某学校来了8位老师，学校将他们分为三个办公室。现在由你来将这8位老师随机分配三个办公室之中
8位老师定义：1～8
办公室定义：［［］，［］，［］］

import random
rooms = [[], [], []]
teachers = [i for i in range(9)]
random.shuffle(teachers)
for i in range(3):
    for j in range(3):
        rooms[i].append(teachers[i * 3 + j])