Python如何从已有用户列表生成随机分组列表 代码实现咨询

核心结论 #

完全可以在现有groups函数内实现随机分组逻辑，不需要推翻现有实现思路，仅需要补充随机打乱、列表拆分的相关代码即可。

现有代码优化点 #

• 不需要循环遍历users统计人数，直接调用len(users)即可拿到参与总人数，简化冗余代码
• 分组数量计算用整数除法//替代浮点数除法/后转整数的写法，逻辑更严谨
• 引入Python内置的random模块即可实现公平的随机排序效果

完整实现代码 #

import random

def participents():
    users = []
    while True:
        participent = input('Who is gonna join? ')
        if participent != 'stop':
            users.append(participent)
        else:
            break
    # 直接获取参与人数，无需循环统计
    counter = len(users)

    def groups():
        # 先随机打乱用户列表，保证分组随机
        random.shuffle(users)
        groups_list = []
        if counter % 3 == 0:
            group_size = 3
            group_num = counter // 3
            print(f"共分为{group_num}个3人小队")
        else:
            group_size = 2
            group_num = counter // group_size
            remainder = counter % group_size
            # 有剩余人数时多算一个组
            if remainder != 0:
                group_num += 1
            print(f"共分为{group_num}个2人小队")
        # 拆分用户列表生成分组
        for i in range(group_num):
            start = i * group_size
            end = start + group_size
            current_group = users[start:end]
            groups_list.append(current_group)
            print(f"第{i+1}组：{','.join(current_group)}")
        return groups_list

    groups()

participents()

逻辑说明 #

• 调用random.shuffle(users)随机打乱用户列表顺序，确保分组无规律
• 按照预设规则优先分配3人组，人数无法被3整除时分配2人组
• 按组大小对打乱后的用户列表进行切片，生成对应的分组列表并打印输出
• 若存在剩余1人的情况，会自动单独成组，也可以根据实际需求调整规则，将剩余成员并入已有队伍

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