实例4：随机分配办公室

某学校新招聘了8名教师，已知该学校有3个空闲办公室且工位充足，现需要随机安排这8名教师的工位。

本实例要求编写程序，将8名教师随机分配到3个办公室中。

提示：随机选择办公室，可以使用random.randint(0,2)实现，需使用import random导入random模块。

实例目标

  1. 掌握列表的嵌套使用

实例分析

本实例中的学校包含3个空闲的办公室，其中的每个办公室可以随意容纳教师。由于学校和办公室分别用于存储办公室与教师，且它们中数据的个数是可变的，因此可以用列表来表示学校和办公室，用嵌套列表表示学校与办公室的包含关系，此时表示第一个办公室的空列表的索引为1，表示第二个办公室的空列表的索引为2，表示第三个办公室的空列表的索引为3。

随机分配办公室就是将每名老师逐个安排到任意的办公室中，这个过程可拆分为两步，第一步就是逐个取出教师姓名，相当于遍历列表元素的操作；第二步就是安排到任意的办公室，相当于使用random模块中randint()方法生成随机0-2之间的整数，将产生的整数作为索引来随机获取嵌套列表的内层列表，之后在该列表中执行添加教师姓名的操作。

代码实现

本实例的具体实现代码如下所示。

import random
# 定义一个列表用来保存3个办公室
offices = [[], [], []]
# 定义一个列表用来存储8位老师的名字
names = ['张老师', '李老师', '赵老师', '高老师',
     '刘老师', '周老师', '王老师', '吴老师']
for name in names:
    # 将8位老师按照索引为0、1、2进行分组
    index = random.randint(0, 2)
    # print(index)
    # 将8位老师放在不同列表中
    offices[index].append(name)
flag = 1
for tempNames in offices:
    print('办公室%d的人数为：%d' % (flag, len(tempNames)))
    flag += 1
    for name in tempNames:
        print("%s" % name, end=' ')
    print(" ")

以上代码首先定义了一个包含3个办公室的嵌套列表offices，定义了另一个包含8位教师姓名的列表names，然后将生成的0~2之间的随机数作为索引获取嵌套列表offices中的任一内层列表，将遍历names取出的元素添加到该内层列表中，直至遍历出最后一个元素为止，最后输出每个办公室的教师分配情况。

需要注意的是，为了美化输出的结果，这里定义了一个表示办公室编号的变量flag，输出每个办公室的总人数与教师姓名。

代码测试

运行程序，程序运行的结果如下所示。

办公室1的人数为：3

张老师 刘老师 王老师

办公室2的人数为：1

赵老师

办公室3的人数为：4

李老师 高老师 周老师 吴老师

再次运行程序，程序运行的结果如下所示。

办公室1的人数为：3

高老师 周老师 吴老师

办公室2的人数为：3

张老师 李老师 刘老师

办公室3的人数为：2

赵老师 王老师