如何高效打乱列表中序列的顺序且保持序列内部元素不变？

问题背景 #

给定一个扁平列表，其中包含多个独立序列：每个序列遵循 x_k, y_k, ..., x_k_i, y_k_i, ... z_k 的模式，以 z_k 作为序列结束标记。需要打乱列表的整体顺序，但每个序列内部的元素顺序必须保持不变。

示例输入：

input_list = ['x_1', 'y_1', 'z_1', 
              'x_2', 'y_2', 'z_2', 
              'x_3_1', 'y_3_1', 'x_3_2', 'y_3_2', 'z_3', 
              'x_4_1', 'y_4_1', 'x_4_2', 'y_4_2', 'x_4_3', 'y_4_3', 'z_4']

符合要求的输出示例（序列整体顺序打乱，内部元素顺序不变）：

shuffled_list = ['x_3_1', 'y_3_1', 'x_3_2', 'y_3_2', 'z_3', 
                 'x_1', 'y_1', 'z_1',
                 'x_4_1', 'y_4_1', 'x_4_2', 'y_4_2', 'x_4_3', 'y_4_3', 'z_4',
                 'x_2', 'y_2', 'z_2']

现有方法的局限性 #

你当前的实现是先将每个序列拆分为独立子列表，形成嵌套列表后随机打乱子列表顺序，再扁平化得到结果。这种方法的问题在于：

• 需要额外内存存储所有子列表，当序列数量多或元素体积大时，内存开销会显著增加
• 子列表的创建和复制过程会带来不必要的性能损耗

更高效的实现方式 #

核心思路是直接定位序列边界，通过切片索引的随机排列来拼接结果，无需创建额外子列表：

import random

input_list = ['x_1', 'y_1', 'z_1', 
              'x_2', 'y_2', 'z_2', 
              'x_3_1', 'y_3_1', 'x_3_2', 'y_3_2', 'z_3', 
              'x_4_1', 'y_4_1', 'x_4_2', 'y_4_2', 'x_4_3', 'y_4_3', 'z_4']

# 1. 定位所有序列的结束索引（找到所有以"z_"开头的元素位置）
end_indices = [idx for idx, item in enumerate(input_list) if item.startswith('z_')]

# 2. 生成每个序列的切片范围（左闭右开）
sequence_slices = []
start_idx = 0
for end_idx in end_indices:
    sequence_slices.append( (start_idx, end_idx + 1) )
    start_idx = end_idx + 1

# 3. 原地打乱切片顺序（无需复制数据）
random.shuffle(sequence_slices)

# 4. 按打乱后的切片顺序拼接原列表的片段
shuffled_list = []
for s, e in sequence_slices:
    shuffled_list.extend(input_list[s:e])

print(shuffled_list)

效率优势 #

1. 内存优化：直接操作原列表的切片索引，无需复制任何序列元素，内存占用仅为原列表加上少量索引存储
2. 性能提升：random.shuffle 是原地高效打乱操作，extend 方法批量添加元素比逐个追加更快
3. 扩展性强：即使处理超大列表，也不会因为子列表的创建导致内存溢出

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