如何用NumPy在Python中随机选择数组的行

在这篇文章中，我们将看到两种不同的方法，如何用NumPy在Python中随机选择数组的行。让我们看看不同的方法，我们可以通过这些方法来选择数组的随机行。

方法1：我们将使用shuffle()函数。shuffle()函数随机地洗掉一个数组的行，然后我们将显示二维数组的随机行。

# import modules
import random
import numpy as np
  
# create 2D array
data = np.arange(50).reshape((5, 10))
  
# display original array
print("Array:")
print(data)
  
# row manipulation
np.random.shuffle(data)
  
# display random rows
print("\nRandom row:")
rows = data[:1, :]
print(rows)

输出:

方法2：首先创建一个数组，然后对其应用sample()方法并显示单行。

# import modules
import random
import numpy as np
  
# create 2D array
data = np.arange(50).reshape((5, 10))
  
# display original array
print("Array:")
print(data)
  
# row manipulation
rows_id = random.sample(range(0, 
                              data.shape[1]-1), 1)
  
# display random rows
print("\nRandom row:")
row = data[rows_id, :]
print(row)

输出:

方法3：我们将使用函数choice()。choices()方法从列表中返回多个随机元素并进行替换。

现在，让我们从我们已经创建的随机整数列表中选择行。

# import modules
import random
import numpy as np
  
# create 2D array
data = np.arange(50).reshape((5, 10))
  
# display original array
print("Array:")
print(data)
  
# row manipulation
number_of_rows = data.shape[0]
random_indices = np.random.choice(number_of_rows, 
                                  size=1, 
                                  replace=False)
  
# display random rows
print("\nRandom row:")
row = data[random_indices, :]
print(row)

输出: