如何在Python中随机将团队列表分配给用户列表？

场景说明 #

我们需要将给定的用户列表随机分配到团队列表中，支持多种分配场景：用户与团队数量一对一、用户多于团队、团队多于用户。

场景1：用户与团队数量一对一 #

如果用户和团队的数量相同，直接将用户随机打乱后与团队配对即可：

import random

# 定义团队和用户列表
teams = ["USA", "England", "India"]
users = ["Mark", "Sean", "Sri"]

# 生成打乱后的用户列表（不修改原列表）
shuffled_users = random.sample(users, len(users))
# 配对团队与随机用户
assignments = list(zip(teams, shuffled_users))

# 输出分配结果
for team, user in assignments:
    print(f"{team}: {user}")

代码解释：

• random.sample() 从用户列表中随机抽取所有元素，生成打乱的新列表，避免修改原数据。
• zip() 将团队列表和打乱后的用户列表按位置配对，得到随机的一对一分配结果。

场景2：用户数量多于团队数量 #

当用户数量比团队多，可将用户随机打乱后，均匀分配到各个团队：

import random

teams = ["USA", "England", "India"]
users = ["Mark", "Sean", "Sri", "Bob", "Alice", "Tom"]

# 直接打乱原用户列表
random.shuffle(users)
# 初始化每个团队的用户列表
assignments = {team: [] for team in teams}

# 循环分配用户到团队
for idx, user in enumerate(users):
    # 通过取模实现循环分配，保证用户分布均匀
    target_team = teams[idx % len(teams)]
    assignments[target_team].append(user)

# 输出分配结果
for team, assigned_users in assignments.items():
    print(f"{team}: {', '.join(assigned_users)}")

代码解释：

• random.shuffle() 原地打乱用户列表。
• 利用 idx % len(teams) 实现循环分配，确保每个团队分到的用户数量尽可能平均。

场景3：团队数量多于用户数量 #

如果团队比用户多，有两种分配方式可选：

方式1：仅随机选择与用户数量匹配的团队进行一对一分配

import random

teams = ["USA", "England", "India", "Japan", "Brazil"]
users = ["Mark", "Sean", "Sri"]

# 随机选择和用户数量相同的团队
selected_teams = random.sample(teams, len(users))
# 配对团队与用户
assignments = list(zip(selected_teams, users))

# 输出结果
for team, user in assignments:
    print(f"{team}: {user}")

方式2：所有团队参与，用户随机分配到任意团队（部分团队可能无用户）

import random

teams = ["USA", "England", "India", "Japan", "Brazil"]
users = ["Mark", "Sean", "Sri"]

# 初始化所有团队的用户列表
assignments = {team: [] for team in teams}

# 随机为每个用户分配团队
for user in users:
    target_team = random.choice(teams)
    assignments[target_team].append(user)

# 输出结果
for team, assigned_users in assignments.items():
    user_str = ', '.join(assigned_users) if assigned_users else '无'
    print(f"{team}: {user_str}")

内容的提问来源于stack exchange，提问作者Sriram Varma