File size: 1,267 Bytes
28e129b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
import os
def get_filenames_from_txt(txt_path):
filenames = set()
with open(txt_path, "r") as f:
for line in f:
file, ext = os.path.splitext(line.strip())
filenames.add(file)
return filenames
def find_difference(txt1_path, txt2_path):
filenames1 = get_filenames_from_txt(txt1_path)
filenames2 = get_filenames_from_txt(txt2_path)
difference1 = filenames1.difference(filenames2)
difference2 = filenames2.difference(filenames1)
return difference1, difference2
def write_to_txt(filename, difference):
with open(filename, "w") as f:
for item in difference:
f.write("%s\n" % item)
if __name__ == "__main__":
txt1_path = "/media/DataDrive/yiling/annotation/XD_test_annotation.txt"
txt2_path = "/media/DataDrive/yiling/annotation/XD_feature_list.txt"
difference1, difference2 = find_difference(txt1_path, txt2_path)
output_txt1 = "/media/DataDrive/yiling/annotation/difference1.txt"
output_txt2 = "/media/DataDrive/yiling/annotation/difference2.txt"
write_to_txt(output_txt1, difference1)
write_to_txt(output_txt2, difference2)
print("txt1中不同的部分已存储到:", output_txt1)
print("txt2中不同的部分已存储到:", output_txt2)
|