Python 按照X,Y样本对打乱数据集

import random

def shuffle(data,label):    
    index = [i for i in range(len(data))] 
    random.shuffle(index) # 需要import
    data = data[index]
    label = label[index] 
    return data,label

上述代码，适用于type(data) 是numpy ，tensor，但不适用于list结构。