从数组 MATLAB Poisson 中提取随机数

Question

这在 MATLAB 中曾经很简单，但在 Python 中仍然如此。我有一组 Lambda，我希望生成泊松分布的随机数以插入直方图中。

lambda = [0.1 0.4]

我以前可以放在那里：

n = poissrnd(a,10)

并为每个 lambda 返回一个 10 行 x 2 列随机生成的泊松数的数组（每列代表来自每个 lambda 的 10 个样本）。出于某种原因，这种情况发生了变化——我不知道为什么。

在 Python 中 - 这仍然可以通过输入：

n = np.random.poisson(lambda, [10,2])

我一直在用头撞桌子 - 任何帮助将不胜感激......

Answer 1

让

N = 10;
lambda = [0.1 0.4];

• 无循环：来自documentation，

  R = poissrnd(lambda) 从平均参数为lambda 的泊松分布生成随机数。 lambda 可以是向量、矩阵或多维数组。 R 的大小就是lambda 的大小。

  所以：

  result = poissrnd(repmat(lambda, N, 1));
  

• 使用for 循环：

  result = NaN(N, numel(lambda)); % preallocate
  for k = 1:numel(lambda)
      result(:,k) = poissrnd(lambda(k), [N 1]);
  end
  

• arrayfun:

  result = cell2mat(arrayfun(@(x) poissrnd(x, [N 1]), lambda, 'uniformoutput', false));