模拟数据

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
{
    "train":[                                           //所有火车信息
    	{
            "trainNumber":"String",                     //火车编号
            "origin":[                                  //该火车某一段起点
                {
                    "place": "String",                  //起点站名称
            		"time":"String",                    //到达起点站时间
                    "end":[                             //以此起点为开头的终点端
                        {
                            "place":"String",           //终点站名称
                            "price":"int",              //该段路程价格
                            "time":"String",            //到达终点站时间
                            "len":"double"              //该路段里程
                        },
                    ]
                },
            ]
    	},
    ]
}

先建图,为图赋权值

随机10000量火车,生成始发站,站数,时间,把站点随机出来

对于每两站对其分析

1
2
3
4
5
6
7
8
9
10
def sj(x, y):                            //已知车型和距离求时间
    if (x == 'K'):
        xx = random.randint(80, 120)
        return y / xx * 60
    if (x == 'D'):
        xx = random.randint(160, 250)
        return y / xx * 60
    if (x == 'G'):
        xx = random.randint(250, 350)
        return y / xx * 60
1
2
3
4
5
6
7
8
def jg(x, y):                         //求价格,数据来源于网上
    if (x == 'K'):
        return 0.1 * y
    elif (x == 'D'):
        return 0.3 * y
    else:
        return 0.46 * y

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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
import random
import json


class edge:
    def __init__(self):
        self.u = ''
        self.v = ''
        self.val = ''


class node:
    def __init__(self):
        self.place = ''
        self.ti = ''
        self.end = []


class endNode:
    def __init__(self):
        self.place = ''
        self.ti = ''
        self.price = ''
        self.len = ''


e = []
lc = [[0 for i in range(1001)] for j in range(1001)]  # 路程


def makeEdge():
    for i in range(1, 100000):
        u = random.randint(1, 1000)
        v = u
        while (u == v):
            v = random.randint(1, 1000)
        #print(u, v)
        if (lc[u][v]):
            continue
        aa = edge()
        aa.val = random.uniform(1, 1000)
        aa.u = u
        aa.v = v
        e.append(aa)
        ab = edge()
        ab.u = aa.v
        ab.v = aa.u
        ab.val = aa.val
        e.append(ab)
        lc[u][v] = aa.val
        lc[v][u] = aa.val


l = 0
cnt = 0
jl = [0] * 100


def sj(x, y):
    if (x == 'K'):
        xx = random.randint(80, 120)
        return y / xx * 60
    if (x == 'D'):
        xx = random.randint(160, 250)
        return y / xx * 60
    if (x == 'G'):
        xx = random.randint(250, 350)
        return y / xx * 60


def jg(x, y):
    if (x == 'K'):
        return 0.1 * y
    elif (x == 'D'):
        return 0.3 * y
    else:
        return 0.46 * y


if __name__ == '__main__':
    makeEdge()
    cnt = 0
    l = len(e)  # e的大小
    train_info = {}
    dat = json.loads(json.dumps(train_info))
    dat["train"] = []
    for i in range(1, 10000):  # 10 0000
        num = random.randint(1000, 10000)
        qnum = random.choice(['K', 'D', 'G'])
        sf = random.randint(1, 1000);  # 始发站
        train_info = {}
        trainTime = [0] * 100
        trainTime[1] = random.randint(1, 43201)
        zs = random.randint(2, 30)
        a = [0] * 1001
        llc = [0] * 100
        cnt = 1
        mq = sf
        a[mq] = 1
        jl[cnt] = mq
        data = json.loads(json.dumps(train_info))
        data['trainNum'] = str(qnum + str(num))
        origin = []
        data['origin'] = []
        for j in range(1, zs):
            a[mq] = 1
            dd=cnt;hh=0
            while(dd == cnt):

                for k in range(0, l):
                    if (e[k].u == mq and a[e[k].v] == 0 and random.randint(1, 1000) % 3 == 0):
                        a[e[k].v] = 1

                        mq = e[k].v;
                        cnt = cnt + 1
                        jl[cnt] = mq
                        llc[cnt] = llc[cnt - 1] + lc[e[k].u][e[k].v]
                        trainTime[cnt] = trainTime[cnt - 1] + int(sj(qnum, lc[e[k].u][e[k].v]))
                        #print(i, mq)

                        break
                hh=hh+1
                if(hh==3):
                    break
        if(cnt==1):
            continue
        for j in range(1, cnt):
            data1 = json.loads(json.dumps(train_info))

            aa = node()
            aa.place = jl[j]
            aa.ti = trainTime[j]
            data1["place"] = aa.place
            data1["time"] = aa.ti
            data1["end"] = []
            #print('#', aa.place, aa.ti)
            for k in range(j + 1, cnt + 1):
                data2 = json.loads(json.dumps(train_info))

                ab = endNode()
                ab.ti = trainTime[k]
                ab.place = jl[k]
                ab.len = llc[k] - llc[j]
                ab.price = jg(qnum, ab.len)
                data2['place'] = ab.place
                data2['time'] = ab.ti
                data2['len'] = ab.len
                data2['price'] = ab.price
                data1["end"].append(data2)
            data['origin'].append(data1)
        dat["train"].append(data);

        article = json.dumps(data, ensure_ascii=False)
        #print(article)
    with open('data.txt', 'w') as f:
        #data = 'some data to be written to the file'
        article = json.dumps(dat, ensure_ascii=False)
        f.write(article)

    #print(article)

生成数据json表示