如何改进Python中学生分组随机划分的代码

import random

def random_grouping(students, num_groups):
    # 随机打乱学生列表
    random.shuffle(students)
    
    # 计算每组的基本人数和剩余人数
    base_size = len(students) // num_groups
    remainder = len(students) % num_groups
    
    groups = []
    start = 0
    
    for i in range(num_groups):
        # 计算当前组的大小
        group_size = base_size + (1 if i < remainder else 0)
        # 切片获取当前组的学生
        group = students[start:start + group_size]
        groups.append(group)
        start += group_size
    
    return groups

# 示例使用
students = ['Alice', 'Bob', 'Charlie', 'David', 'Eva', 'Frank', 'Grace', 'Hank']
num_groups = 3
groups = random_grouping(students, num_groups)
for i, group in enumerate(groups):
    print(f"Group {i+1}: {group}")