numpy: zufällige Auswahl von K Elementen von insgesamt M (M> K) Elementen?

Question

Gibt es eine praktische Implementierung der Matlab-Funktion randperm in numpy, die zufällig K-Elemente von total M (M> K) Elementen auswählt und die ausgewählte Indice zurückgibt?

In Matlab,

randperm(100,10) 

ans = 

82 90 13 89 61 10 27 51 97 88 

Answer 1

Ja, mit dem numpy.random.choice function.

>>> numpy.random.choice(100, 10, replace=False) 
array([89, 99, 27, 39, 80, 31, 6, 0, 40, 93]) 

Beachten Sie, dass der resultierende Bereich 0 bis M-1 ist. Wenn Sie 1 bis M wie MATLAB benötigen, fügen 1 das Ergebnis:

>>> numpy.random.choice(100, 10, replace=False) + 1 
array([ 28, 23, 15, 90, 18, 65, 86, 100, 99, 1])