我正在研究一个简单的问题,我想编写一个函数,该函数将从二项分布中随机抽取n个大小为q的样本(例如60%1和40%0),然后保存每个中1s的份额在数组中采样(以便以后可以分析该数组)。例如,我想模拟一下,如果我从60%1s和40%0s的分布中抽取1,000个大小为30的样本,我想将长度为1,000的数组作为输出,其中每行代表该样本中的1。
您的意思是伯努利分布吗?
from scipy.stats import bernoulli
data = [bernoulli.rvs(0.6, size=30) for _ in range(1000)]