[Numpy-discussion] Re: selecting random array element
Robert Kern
robert.kern at gmail.com
Mon Feb 20 15:47:01 EST 2006
Chris Fonnesbeck wrote:
> What is the best way to select a random element from a numpy array? I
> know I could index by a random integer, but was wondering if there was
> a built-in method or function.
Generating a random index is what I do.
I think there's certainly room for a RandomState.choice() method. I think
something like this covers most of the use cases:
import numpy
from numpy import random
def choice(x, axis=None):
"""Select an element or subarray uniformly randomly.
If axis is None, then a single element is chosen from the entire array.
Otherwise, a subarray is chosen from the given axis.
"""
x = numpy.asarray(x)
if axis is None:
length = numpy.multiply.reduce(x.shape)
n = random.randint(length)
return x.flat[n]
else:
n = random.randint(x.shape[axis])
# I'm sure there's a better way of doing this
idx = map(slice, x.shape)
idx[axis] = n
return x[tuple(idx)]
--
Robert Kern
robert.kern at gmail.com
"In the fields of hell where the grass grows high
Are the graves of dreams allowed to die."
-- Richard Harter
More information about the NumPy-Discussion
mailing list