get_intra_group.py

import os
import numpy as np
import codecs
import sys
root_path = sys.argv[1]
dirs = os.listdir(root_path)
print(dirs)
for filename in dirs:
    file_path = root_path + "/" + filename
    f = open(file_path, "r")
    output_path = root_path + "/" + filename + "." + "tree"
    output = codecs.open(output_path, 'w', encoding='utf-8')
    for line in f.readlines():
        line = line.strip("\n")
        line = line.split(" ")
        mapping = ""
        idx = 1
        for word in line:
            if word[-2:] == "@@":
               mapping+=(str(idx))
               mapping += " "
               continue
            else:
               mapping += str(idx)
               mapping += " "
               idx += 1
               continue
        mapping = mapping[:-1]
        mapping += "\n"
        output.write(mapping)