How to randomly select rows of an array in Python with NumPy ?
In this article, we will see two different methods on how to randomly select rows of an array in Python with NumPy. Let’s see different methods by which we can select random rows of an array:
Method 1: We will be using the function shuffle(). The shuffle() function shuffles the rows of an array randomly and then we will display a random row of the 2D array.
Python3
import random
import numpy as np
data = np.arange( 50 ).reshape(( 5 , 10 ))
print ( "Array:" )
print (data)
np.random.shuffle(data)
print ( "\nRandom row:" )
rows = data[: 1 , :]
print (rows)
|
Output:
Method 2: First create an array, then apply the sample() method to it and display a single row.
Python3
import random
import numpy as np
data = np.arange( 50 ).reshape(( 5 , 10 ))
print ( "Array:" )
print (data)
rows_id = random.sample( range ( 0 ,
data.shape[ 1 ] - 1 ), 1 )
print ( "\nRandom row:" )
row = data[rows_id, :]
print (row)
|
Output:
Method 3: We will be using the function choice(). The choices() method returns multiple random elements from the list with replacement.
Now lets, select rows from the list of random integers that we have created.
Python3
import random
import numpy as np
data = np.arange( 50 ).reshape(( 5 , 10 ))
print ( "Array:" )
print (data)
number_of_rows = data.shape[ 0 ]
random_indices = np.random.choice(number_of_rows,
size = 1 ,
replace = False )
print ( "\nRandom row:" )
row = data[random_indices, :]
print (row)
|
Output:
Last Updated :
25 Feb, 2021
Like Article
Save Article
Share your thoughts in the comments
Please Login to comment...