Agent-Ticket/OrderGenerate.py at master · Hyperion-shuo/Agent-Ticket · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import numpy as np


def OrderGenerator(data, mode):
    """
    :param data: 一个起飞日的87天价格list
    :param mode: 1，2 两个难度，2更难
    :return: 长为87的订单list 有订单的日期对应数值为订单价格， 没订单的日期对应-1

    订单的日期分布是每天1/3概率产生新订单
    """

    order_list = []
    order_price = -1
    if mode == 1:

        for i in range(87):
            have_order = np.random.randint(0, 3)
            if have_order == 0:
                order_price = np.min(np.hstack((np.average(data[:i+1]), data[i]))) - np.random.randint(0, 50)
                # order_price = 1877
            else:
                order_price = -1

            # temp_day = i
            # sum_price_after = 0
            # lowest_price_after = data[i]
            # while temp_day <= 87:
            #     today_price = data[i]
            #     sum_price_after += today_price  # avg_price 包含今天的价格因为可以选择当天买票
            #     if today_price < lowest_price_after:
            #         lowest_price_after = today_price
            #     temp_day += 1
            # # sum_price 有today_price 因此分母要加1
            # avg_price_after = sum_price_after / (87 - i + 1)
            # # 读取订单价格、未来最低价，计算最大收益
            # avgProfit = order_price - avg_price_after
            # maxProfit = order_price - lowest_price_after

            order_list.append(order_price)

        return order_list


    elif mode == 2:
        for i in range(87):
            have_order = np.random.randint(0, 3)
            if have_order == 0:
                order_price = np.min(np.hstack((np.average(data[:i+1]), data[i]))) * ((0.95 - 0.7) * np.random.sample() + 0.7)
            else:
                order_price = -1
            order_list.append(order_price)

    return order_list


def readRoute(filename):
    with open(filename, 'r') as f:
        content = f.readlines()
        result = []
        for c in content:
            list = []
            c = c.strip("[]\n")
            c = c.split(",")
            for i in range(89):
                if i < 87:
                    list.append(float(c[i]))
                else:
                    c[i] = c[i].replace("'", "")
                    list.append(c[i].replace(" ", ""))
            result.append(list)
        # print(result)
        return result

if __name__ == "__main__":
    # route_list = readRoute("./wang/data/route")
    # print(len(OrderGenerator(route_list[21],1)))
    # print(route_list[21][:0])
    a = [1,2,3,4,5,6,7,8,9]
    a = [s**2  for s in a  if s>5]
    print(a)