如何读取brat注解toll提供的ann文件并将其转换为python中的dataframe？

pip install pandas

T1  实体类型 0 5    实体文本

R1  关系类型 Arg1:T1 Arg2:T2

import pandas as pd

def parse_ann_file(file_path):
    entities = []
    relations = []
    
    with open(file_path, 'r', encoding='utf-8') as file:
        for line in file:
            parts = line.strip().split('\t')
            if len(parts) == 3:  # 这是一个实体行
                identifier, entity_info, text = parts
                entity_type, start, end = entity_info.split(' ')
                entities.append({
                    'ID': identifier,
                    'Type': entity_type,
                    'Start': int(start),
                    'End': int(end),
                    'Text': text
                })
            elif len(parts) == 4:  # 这是一个关系行
                identifier, relation_info, arg1, arg2 = parts
                relation_type, arg1_id, arg2_id = relation_info.split(' ')
                relations.append({
                    'ID': identifier,
                    'Type': relation_type,
                    'Arg1': arg1_id,
                    'Arg2': arg2_id
                })
    
    entities_df = pd.DataFrame(entities)
    relations_df = pd.DataFrame(relations)
    
    return entities_df, relations_df

# 使用函数
file_path = 'path_to_your_ann_file.ann'
entities_df, relations_df = parse_ann_file(file_path)

print(entities_df.head())
print(relations_df.head())