Python随机取一个矩阵数组的某几行

最新推荐文章于 2024-11-05 19:39:41 发布

原创最新推荐文章于 2024-11-05 19:39:41 发布 · 1.2w 阅读

15 ·

CC 4.0 BY-SA版权

Python 专栏收录该内容

36 篇文章

订阅专栏

本文介绍如何利用Python的NumPy库创建数组，并演示了如何通过循环增加元素来构建一个特定的二维数组。此外，还展示了如何使用NumPy的随机模块对数组进行随机重排，提供了两组随机重排后的数组实例。

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]]