如何在 MATLAB 中从 N 个点中随机选择 k 个点？

Question

我使用此代码创建和绘制N 点：

N=input('No. of Nodes:');
data = rand(N,2) % Randomly generated n no. of nodes
x = data(:,1);
y = data(:,2);
plot(x,y,'*');

如何从N 点中选择k 点（概率为p=0.25），然后将那些k 点涂成红色并将其他点保留为*。

Answer 1

您可以采取两种方法。第一个解决方案是从N 值中随机选择k 值，这将确保您始终选择k 点。第二种解决方案是随机选择值，每个值的平均概率为p，这可能导致随机选择少至0 或多至N。

• 从N 值中选择k：

  您可以使用函数RANDPERM 创建整数1 到N 的随机排列，然后选择排列列表中的第一个k 值并将它们重新绘制为红色：

  index = randperm(N);
  plot(x(index(1:k)),y(index(1:k)),'r*');
  

• 以平均概率选取值p：

  您可以使用RAND 函数为您的每个N 值从0 到1 中选择一个随机值，然后选择随机值小于或等于您的平均概率@ 987654340@ 并将它们重新绘制为红色：

  index = (rand(N,1) <= p);
  plot(x(index),y(index),'r*');
  

Answer 2

据我了解，对于每个N 随机点，您要掷硬币来决定是否选择它（其中硬币的成功概率为p=0.25！）

data = rand(N,2);             %# generate random points
index = (rand(N,1) <= p);     %# roll coins to pick with prob p
data(~index, :) = [];         %# keep only selected points

这最终相当于首先只生成p*N 随机点（至少随着N 变大，您会接近这个数字）...

data = rand(p*N, 2);          %# directly generate p*N number of points

您可以针对各种 N 值测试最后一条语句：

fprintf('1st = %d \n', p*N)
fprintf('2nd = %d \n', sum(rand(N,1) <= p))