如何在Python中生成每行包含指定范围所有数字的随机矩阵？

直接用random.randint(0,5)生成行的话，很容易出现数字缺失的问题，核心思路是先确保每行包含0到5的全部数字，再补充随机数，下面是两种可行的实现方式：

方法一：先填充必选数字，再补随机数（推荐） #

这种方法效率高，不会出现无限循环的情况，步骤如下：

• 先把0-5这6个数字打乱顺序，作为行的基础内容
• 根据需要的行长度，补充若干个0-5的随机数
• 最后把基础内容和补充的随机数再次打乱，保证整体顺序随机

代码示例：

import random

def generate_valid_row(row_length):
    # 初始化必须包含的0-5数字并打乱
    required_nums = list(range(6))
    random.shuffle(required_nums)
    
    # 计算需要补充的随机数数量，行长度不能小于6
    extra_count = row_length - 6
    if extra_count < 0:
        raise ValueError("行长度不能小于6，需包含0-5所有数字")
    
    # 生成补充的随机数
    extra_nums = [random.randint(0, 5) for _ in range(extra_count)]
    
    # 合并后再次打乱
    full_row = required_nums + extra_nums
    random.shuffle(full_row)
    return full_row

# 生成3行、每行长度为8的矩阵
matrix = [generate_valid_row(8) for _ in range(3)]
for row in matrix:
    print(row)

方法二：循环生成直到符合要求（不推荐） #

这种方式是先随机生成一行，再检查是否包含所有0-5的数字，不符合就重新生成。优点是逻辑简单，但行长度越短，重复生成的概率越高，效率较低。

代码示例：

import random

def generate_valid_row(row_length):
    while True:
        row = [random.randint(0, 5) for _ in range(row_length)]
        # 用集合判断是否包含全部数字
        if set(row) == set(range(6)):
            return row

# 生成矩阵
matrix = [generate_valid_row(8) for _ in range(3)]
for row in matrix:
    print(row)

内容的提问来源于stack exchange，提问作者Nataša Radukić