题解 | 数据集随机洗牌

import re
import numpy as np

def shuffle_data(X, y, seed=None):
    if seed:
        np.random.seed(seed)
    
    idx = np.random.permutation(len(X))
    return X[idx], y[idx]
    
if __name__ == "__main__":
    X = np.array(eval(input()))
    y = np.array(eval(input()))
    print(shuffle_data(X, y, 42))