Fastest way to get elements from a numpy array and create a new numpy array

Question

I have numpy array called data of dimensions 150x4 I want to create a new numpy array called mean of dimensions 3x4 by choosing random elements from data.

My current implementation is:

    cols = (data.shape[1])
    K=3
    mean = np.zeros((K,cols))
    for row in range(K):
        index = np.random.randint(data.shape[0])
        for col in range(cols):
            mean[row][col] = data[index][col]

Is there a faster way to do the same?

ysakamoto · Accepted Answer · 2014-05-11 02:36:41Z

2

You can specify the number of random integers in numpy.randint (third argument). Also, you should be familiar with numpy.array's index notations. Here, you can access all the elements in one row by : specifier.

mean = data[np.random.randint(0,len(data),3),:]

edited May 11, 2014 at 2:36

answered May 11, 2014 at 2:31

ysakamoto

2,5421 gold badge18 silver badges22 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Fastest way to get elements from a numpy array and create a new numpy array

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related