python打乱带标签的数据集简便方法

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索引重新组合