예제 #1
0
 def read_data(self, name = 'train.json'):
     self.data = Json.load(os.path.join(self.file_path, name))
예제 #2
0
            print(i, len(ret))
            username = users[i]
            traj_path = os.path.join(self.file_path, username, 'Trajectory')
            trajs = os.listdir(traj_path)
            for j in range(len(trajs)):
                ret_traj = self.generate_json_single(
                    os.path.join(traj_path, trajs[j]))
                for k in ret_traj:
                    if len(k) < 10:
                        continue
                    ret[str(cnt)] = k
                    cnt += 1
                    # print(ret[str(cnt)])
        Json.output(ret, os.path.join(self.file_path, name))
        return len(ret)


if __name__ == '__main__':
    file_path = '/home/yiwei/data/Geolife1.3/Data'
    # f = geolife(file_path)
    # f.generate_json_all('all.json')
    dic = {}
    cnt = 0
    js = Json.load('/home/yiwei/data/Geolife1.3/Data/small.json')
    for k, v in js.items():
        if (int(k) % 10 != 0):
            print(k)
            dic[str(cnt)] = [[float(x[0]), float(x[1]), x[2]] for x in v]
            cnt += 1
    Json.output(dic, '/home/yiwei/data/Geolife1.3/Data/small.json')