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