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