def load_wifi_data_new(self, dir_path, save_path="wifi_floor_temp.npz", ref_list=[]): load_in = LoadWifiData() file_list = os.listdir(dir_path) floor_data = {} for floor in file_list: print floor fl_path = os.path.join(dir_path, floor) if os.path.isfile(fl_path): continue floor_file_list = os.listdir(fl_path) for file_name in floor_file_list: if file_name.endswith("wifi"): name, ext = os.path.splitext(file_name) print name wifi_file = name + ".wifi" wifi_file = os.path.join(fl_path, wifi_file) floor_data[floor] = load_in.extract_wifi(wifi_file, ref_list) t_bind = [] for x in floor_data: t_bind += floor_data[x].wifi_list wifi_bld_list = list(set(t_bind)) TD = 0 for x in floor_data: D = floor_data[x].wifi_matrix.shape[0] TD += D L = len(wifi_bld_list) t_mat = zeros((D, L)) for i, j in enumerate(floor_data[x].wifi_list): ind = wifi_bld_list.index(j) t_mat[:, ind] = floor_data[x].wifi_matrix[:, i] floor_data[x].wifi_bld_mat = t_mat savez(save_path, floor=floor_data, wifilist=wifi_bld_list, entry_num=TD) return floor_data, wifi_bld_list, TD