怎么创创建一个集合储存员工信息,员工信息包括工号、姓名、年龄、工资等。要求员工信息中不允许出现同工号的员工
时间: 2024-01-29 13:02:20 浏览: 66
你可以使用Python中的集合(set)来储存员工信息,并使用字典(dict)来表示每个员工的工号、姓名、年龄、工资等信息。具体的代码实现如下:
```python
# 创建一个空集合
employee_set = set()
# 添加员工信息
def add_employee(employee_id, name, age, salary):
# 判断员工是否已存在
for employee in employee_set:
if employee['employee_id'] == employee_id:
print('员工已存在')
return
# 创建新员工信息字典
new_employee = {'employee_id': employee_id, 'name': name, 'age': age, 'salary': salary}
# 将新员工信息添加到集合中
employee_set.add(new_employee)
print('员工添加成功')
# 测试代码
add_employee('001', '张三', 25, 5000)
add_employee('002', '李四', 30, 8000)
add_employee('001', '王五', 28, 6000) # 这里会提示员工已存在
```
在上面的代码中,我们首先创建了一个空集合employee_set来储存员工信息。然后定义了一个add_employee函数来添加员工信息。在添加之前,我们先遍历集合中的所有员工信息,判断是否已存在同工号的员工。如果存在,则直接返回,否则创建一个新的员工信息字典,并将其添加到集合中。
最后,我们对代码进行了简单的测试,添加了两个员工信息,其中一个员工的工号与已有员工相同,导致添加失败。
阅读全文