解决方法将用户ID划分为不同组别的一种常用方法是使用字典(dictionary)来存储用户ID和其所属组别。下面是一个示例代码:
# 定义用户ID列表
user_ids = [1001, 1002, 1003, 1004, 1005, 1006, 1007, 1008, 1009, 1010]
# 定义用户ID与组别的对应关系
group_mapping = {
1001: 'A',
1002: 'A',
1003: 'B',
1004: 'B',
1005: 'C',
1006: 'C',
1007: 'D',
1008: 'D',
1009: 'D',
1010: 'E'
}
# 创建一个空字典用于存储划分结果
grouped_users = {}
# 遍历用户ID列表,将用户ID划分到不同组别中
for user_id in user_ids:
group = group_mapping.get(user_id) # 获取用户ID对应的组别
if group:
if group not in grouped_users:
grouped_users[group] = [] # 如果组别不存在,则创建一个空列表
grouped_users[group].append(user_id) # 将用户ID添加到对应组别的列表中
# 打印划分结果
for group, users in grouped_users.items():
print(f"Group {group}: {users}")
这段代码首先定义了用户ID列表user_ids
,以及用户ID与组别对应关系的字典group_mapping
。然后,创建一个空的字典grouped_users
用于存储划分结果。
接下来,通过遍历用户ID列表,使用group_mapping.get(user_id)
获取用户ID对应的组别,然后将用户ID添加到对应组别的列表中。
最后,通过遍历grouped_users
字典,打印出划分结果。
上一篇:按用户ID过滤松果向量数据库。