【matlab】以一定的概率生成随机数

问题：今天在研究中需要以一定的概率产生随机数，要生成一个矩阵c，行标和列标一致时为零，其他情况以概率为p生成[1,10]的正整数，以概率为（1-p）生成0。

 

找到了一个很好的函数：

randsrc(100,1,[alphabet; prob])

alphabet表示需要产生的数字，prob表示对应的概率，注意：两者的维数要一致。

比如：要产生 1、4、 6这三个数。它们分别出现的概率为 0.1 、0.4、 0.5 如何设计程序使得按照这个概率产生100个随机数呢？

alphabet = [1 4 6]; prob = [0.1 0.4 0.5];

randsrc(100,1,[alphabet; prob])

 

对于我的问题，我的想法是当角标一样时，赋值为0，当不一样时，生成一个[1,10]的正整数a，与0组合成一个新的矩阵b，此时以一定的概率添加到c中。

当c的大小为5*5时，程序为：

clc

clear all

n=5;

p=30;

prob=[0.3,1-p/100];

c=[];

for i=1:n

    for j=1:n

        if i==j

            c(i,j)=0;

        else

            a=round(rand(1,1)*9+1);

            b=[a 0];

            c(i,j)=randsrc(1,1,[b;prob]);

        end

    end

end

 

问题得到解决。