numpy利用下标打乱数据集

Posted 2022-09-27 redo19990701

利用下标打乱数据集

import numpy as np
#导入numpy库
def shuffleData(data):
    index=np.arange(len(data))
    #生成一个数据集行数大小的顺序数组[0,1,...,len(data)](顺序下标)
    np.random.shuffle(index)
    #打乱顺序下标
    data=data[index]
    #按照乱序下标重新排列数据集
    return data
data=np.arange(18)
data.resize((6,3))
#生成一个4*2的数据集
print("orign:\n",data)
data=shuffleData(data)
print("Shuffled:\n",data)
#输出:
# orign:
#  [[ 0  1  2]
#  [ 3  4  5]
#  [ 6  7  8]
#  [ 9 10 11]
#  [12 13 14]
#  [15 16 17]]
# Shuffled:
#  [[ 0  1  2]
#  [15 16 17]
#  [ 3  4  5]
#  [ 9 10 11]
#  [ 6  7  8]
#  [12 13 14]]