需要创建一个集合储存员工信息,员工信息包括工号、姓名、年龄、工资等。要求员工信息中不能出现相同工号的员工。
时间: 2024-05-02 12:23:12 浏览: 95
可以使用Python的集合类型(set)来储存员工信息,代码如下:
```python
employees = set() # 创建一个空集合来储存员工信息
# 添加员工信息
def add_employee(emp_id, name, age, salary):
employee = (emp_id, name, age, salary) # 将员工信息封装为一个元组
if emp_id not in [e[0] for e in employees]: # 判断员工信息中是否已有相同工号的员工
employees.add(employee) # 如果没有相同工号的员工,则将员工信息添加到集合中
else:
print("该工号已存在,请重新输入!")
# 测试
add_employee("001", "张三", 25, 5000)
add_employee("002", "李四", 30, 8000)
add_employee("001", "王五", 28, 6000) # 工号为"001"的员工已存在,无法添加
add_employee("003", "赵六", 35, 10000)
# 输出集合中的员工信息
for emp in employees:
print("工号:{}, 姓名:{}, 年龄:{}, 工资:{}".format(emp[0], emp[1], emp[2], emp[3]))
```
运行结果如下:
```
该工号已存在,请重新输入!
工号:001, 姓名:张三, 年龄:25, 工资:5000
工号:002, 姓名:李四, 年龄:30, 工资:8000
工号:003, 姓名:赵六, 年龄:35, 工资:10000
```
阅读全文