from entity import *
import random
import numpy as np

def get_data():
    data = load("train_data_02.pkl")
    new_data = [[] for _ in range(144)]
    for d in data:
        for match in d:
            if match.order.arrive_time >= match.order.order_time or match.order.order_time >= 142:
                new_data[match.order.order_time-1].append(match)

    for i in new_data:
        print(len(i))
    save(new_data,"train_data02.pkl")

if __name__ == '__main__':
    get_data()