随机分配办公室--课后程序(Python程序开发案例教程-黑马程序员编著-第4章-课后作业)

实例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

赵老师 王老师