以下是一个示例代码,演示如何按照角色筛选用户:
# 定义用户类
class User:
def __init__(self, name, role):
self.name = name
self.role = role
# 定义用户列表
users = [
User("Alice", "admin"),
User("Bob", "user"),
User("Charlie", "admin"),
User("David", "user"),
User("Eve", "guest")
]
# 定义角色筛选函数
def filter_users_by_role(users, role):
filtered_users = []
for user in users:
if user.role == role:
filtered_users.append(user)
return filtered_users
# 调用筛选函数,按照角色筛选用户
filtered_users = filter_users_by_role(users, "admin")
# 输出筛选结果
for user in filtered_users:
print(user.name)
以上代码定义了一个User
类,每个用户对象有一个名字和角色属性。然后创建了一个用户列表users
,其中包含了不同角色的用户。
接下来定义了一个filter_users_by_role
函数,该函数接收一个用户列表和一个角色作为参数,然后遍历用户列表,将符合指定角色的用户添加到一个新的列表中,并返回该列表。
最后,调用filter_users_by_role
函数,传入用户列表和需要筛选的角色,得到筛选结果并输出。在示例代码中,我们按照"admin"角色筛选用户,输出结果为"Alice"和"Charlie"。