创建一个集合储存员工信息,员工信息包括工号、姓名、年龄、工资等。要求员工信息中不允许出现同工号的员工。
时间: 2024-05-05 18:22:54 浏览: 159
可以使用Python的set集合来储存员工信息,每个员工信息可以用一个字典来表示,字典的键值对分别表示工号、姓名、年龄、工资等。在添加新的员工信息时,需要先判断该员工的工号是否已经存在于集合中,如果存在则不允许添加新的员工信息。
示例代码如下:
```python
employees = set()
def add_employee(emp_id, name, age, salary):
# 判断工号是否已经存在
for emp in employees:
if emp['emp_id'] == emp_id:
print('Error: employee with the same ID already exists!')
return
# 如果不存在则添加新的员工信息
employees.add({
'emp_id': emp_id,
'name': name,
'age': age,
'salary': salary
})
# 添加新的员工信息
add_employee(1001, '张三', 30, 5000)
add_employee(1002, '李四', 25, 4000)
add_employee(1001, '王五', 28, 4500) # 这里会出现错误,因为工号1001已经存在
# 打印所有员工信息
for emp in employees:
print(emp)
```
输出结果:
```
Error: employee with the same ID already exists!
{'emp_id': 1002, 'name': '李四', 'age': 25, 'salary': 4000}
{'emp_id': 1001, 'name': '张三', 'age': 30, 'salary': 5000}
```
阅读全文