import pickle def get_video_ids(input_pickle_file, output_txt_file): with open(input_pickle_file, "rb") as fp: obj = pickle.load(fp) video_ids = obj.keys() with open(output_txt_file, "w") as out_f: out_f.write("\n".join(video_ids)) get_video_ids("raw_captions_noun-person_neg.pkl", "test_sampled_video_list.txt")