import numpy as np
data = np.array([[1,2],[4,5],[3,6],[7,8]])
label = np.array([1,0,1,1])
data_size = data.shape[0] # 数据集个数
arr = np.arange(data_size) # 生成0到datasize个数
np.random.shuffle(arr) # 随机打乱arr数组
data = data[arr] # 将data以arr索引重新组合
label = label[arr] # 将label以arr索引重新组合
如果出现如下错误,那么可能是因为data不是np对象,而是数组:
则把代码改为如下即可:
data = np.asarray(data)[arr] # 将data以arr索引重新组合
label = label[arr] # 将label以arr索引重新组合