def read_data(self, name = 'train.json'): self.data = Json.load(os.path.join(self.file_path, name))
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')