如何在随机0-1矩阵生成函数中集成指定1的密度参数d？

首先你的初始代码存在逻辑错误：嵌套的while和for循环会生成 n*m 行数据，而非预期的 n 行。先修正这个问题，再集成密度参数 d（d 取值范围建议为0到1），下面提供两种可靠实现方式：

方式一：按概率生成（每个元素独立） #

这种方式下，每个位置以概率 d 生成1，1-d 生成0，整体密度会趋近于 d，适合对密度精度要求不是极高的场景：

import random

def create_matrix(n, m, d):
    grid = []
    for _ in range(n):
        # 每行生成m个元素，每个元素以概率d取1，否则取0
        row = [1 if random.random() < d else 0 for _ in range(m)]
        grid.append(row)
    return grid

方式二：精确控制1的数量 #

如果需要严格保证1的数量为 n*m*d（四舍五入取整），可以先创建全0矩阵，再随机挑选对应数量的位置设为1：

import random

def create_matrix(n, m, d):
    # 计算需要的1的总数，四舍五入取整
    total_ones = round(n * m * d)
    # 创建全0矩阵
    grid = [[0 for _ in range(m)] for _ in range(n)]
    # 生成所有位置的坐标对
    all_positions = [(i, j) for i in range(n) for j in range(m)]
    # 随机挑选total_ones个位置
    selected_positions = random.sample(all_positions, total_ones)
    # 将选中的位置设为1
    for i, j in selected_positions:
        grid[i][j] = 1
    return grid

补充说明 #

• 方式一实现简单、性能较好，适合大矩阵场景；
• 方式二能精确控制1的数量，适合对密度精度要求高的场景；
• 可以给函数加参数校验，比如 if not 0 <= d <= 1: raise ValueError("d must be between 0 and 1")，避免非法输入。

内容的提问来源于stack exchange，提问作者João Garcia