如何从二维数组的每个子数组中随机抽取元素组成新数组

核心逻辑没有复杂门槛，最优实现的时间复杂度为O(k)，其中k是二维数组包含的子数组总数——你必须遍历每个子数组恰好一次，不存在比这个复杂度更低的方案。
具体执行步骤：

• 顺序遍历外层二维数组的每一个子数组
• 对当前子数组，随机选取其中一个元素
• 按子数组的遍历顺序，把选到的元素收集成新数组返回

Python 代码实现 #

直接使用标准库的随机选择接口即可，不需要自己实现随机逻辑：

import random

# 你的原始二维数组
source_arr = [
    [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
    [0, 1, 2, 3, 4, 5],
    [0, 1, 2, 3, 4, 5, 6, 7]
]

def random_pick(arr_2d):
    return [random.choice(sub_arr) for sub_arr in arr_2d]

调用效果和你给出的示例完全一致：

# 第一次调用
print(random_pick(source_arr))  # 可能输出 [2, 5, 6]
# 第二次调用
print(random_pick(source_arr))  # 可能输出 [8, 0, 7]

如果你需要生成多组结果，重复调用random_pick函数即可，每次调用都是独立的随机采样，不会保留上一次的状态。

注意：不要用先打乱整个子数组再取第一个元素的方式实现，那种方式的时间复杂度是O(m)（m是子数组的长度），比直接随机选一个元素的O(1)操作效率低，尤其是子数组很长的时候性能差距会很明显。

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