train_test_split 分层抽样

 train_test_split有参数stratify

比如stratify = X['product_no']，可以按希望的字段进行分层抽样，参数接收的类型是array-like