1. 程式人生 > 程式設計 >Python實現隨機取一個矩陣陣列的某幾行

Python實現隨機取一個矩陣陣列的某幾行

廢話不多說了,直接上程式碼吧!

import numpy as np
 
array = np.array([0,0])
for i in range(10):
  array = np.vstack((array,[i+1,i+1]))
print(array)
# [[ 0 0]
# [ 1 1]
# [ 2 2]
# [ 3 3]
# [ 4 4]
# [ 5 5]
# [ 6 6]
# [ 7 7]
# [ 8 8]
# [ 9 9]
# [10 10]]
 
rand_arr = np.arange(array.shape[0])
 
np.random.shuffle(rand_arr)
print(array[rand_arr[0:5]])
# [[9 9]
# [4 4]
# [1 1]
# [5 5]
# [8 8]]
np.random.shuffle(rand_arr)
print(array[rand_arr[0:5]])
# [[10 10]
# [ 3 3]
# [ 4 4]
# [ 8 8]
# [ 5 5]]

以上這篇Python實現隨機取一個矩陣陣列的某幾行就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。