#!/usr/bin/env python3 # import sys, json import ntpath list_of_sph = sys.argv[1] jsonfile = sys.argv[2] out_partition_json = sys.argv[3] list_of_sph = [line.rstrip("\n") for line in open(list_of_sph)] sph_basename_list = [] for f in list_of_sph: bsname = ntpath.basename(f) sph_basename_list.append(ntpath.basename(f)) json_str = [line.rstrip("\n") for line in open(jsonfile)] num_json = len(json_str) out_partition = open(out_partition_json, "w", encoding="utf-8") for i in range(num_json): if json_str[i] != "": cur_json = json.loads(json_str[i]) cur_rec = cur_json["recording_id"] cur_sph_basename = cur_rec + ".sph" if cur_sph_basename in sph_basename_list: out_json_line = json_str[i] out_partition.write(out_json_line) out_partition.write("\n")