创建一个集合储存员工信息,员工信息包括工号、姓名、年龄、工资等。要求员工信息中不允许出现同工号的员工。
时间: 2024-05-01 22:20:27 浏览: 115
可以使用Python中的set数据结构来储存员工信息,每个元素是一个字典,包含工号、姓名、年龄、工资等信息。可以通过判断工号是否已经存在来实现不允许出现同工号的员工。
具体实现如下:
```python
employee_set = set()
def add_employee(employee):
for emp in employee_set:
if emp['工号'] == employee['工号']:
print('该工号已存在,无法添加员工')
return
employee_set.add(employee)
print('成功添加员工', employee)
employee1 = {'工号': '001', '姓名': '张三', '年龄': 25, '工资': 5000}
employee2 = {'工号': '002', '姓名': '李四', '年龄': 30, '工资': 6000}
employee3 = {'工号': '001', '姓名': '王五', '年龄': 28, '工资': 5500}
add_employee(employee1) # 成功添加员工 {'工号': '001', '姓名': '张三', '年龄': 25, '工资': 5000}
add_employee(employee2) # 成功添加员工 {'工号': '002', '姓名': '李四', '年龄': 30, '工资': 6000}
add_employee(employee3) # 该工号已存在,无法添加员工
```
在上述代码中,我们定义了一个全局变量`employee_set`来存储员工信息,然后定义了一个`add_employee`函数来添加员工。在添加员工之前,我们通过遍历`employee_set`中的每个员工,判断待添加的员工的工号是否已经存在。如果存在,则输出提示信息并返回,否则将员工添加到`employee_set`中,并输出成功添加的员工信息。最后,我们创建了三个员工信息,其中第一个和第三个员工的工号相同,通过调用`add_employee`函数来添加员工,可以看到只有前两个员工被成功添加到`employee_set`中。
阅读全文