| def bio_2_json_one(anno_txt):
|
| ls = anno_txt.split('\n')
|
| text = ''
|
| anno = []
|
| now_label = ''
|
| for i, l in enumerate(ls):
|
| char, label = l.split('\t')
|
| text += char
|
| if 'B-' in label:
|
| start = i
|
| now_label = label.split('-')[1]
|
| if label == 'O':
|
| if now_label:
|
| anno.append([start, i, text[start:i], now_label])
|
| now_label = ''
|
| start = 0
|
| if now_label:
|
| i += 1
|
| anno.append([start, i, text[start:i], now_label])
|
| return {'text': text, 'anno': anno}
|
|
|
|
|
| def bit_2_json(txt):
|
| anno_txts = txt.split('\n\n')
|
| annos = []
|
| for anno_txt in anno_txts:
|
| if anno_txt == '':
|
| continue
|
| anno_j = bio_2_json_one(anno_txt)
|
| annos.append(anno_j)
|
| return annos
|
|
|
|
|
| if __name__ == '__main__':
|
| txt = '''你\tB-PER
|
| 是\tO
|
| 一\tO
|
| 个\tO
|
| 聪\tB-PER
|
| 明\tI-PER
|
| 的\tO
|
| 软\tB-ORG
|
| 件\tI-ORG
|
| 工\tI-ORG
|
| 程\tI-ORG
|
| 师\tI-ORG'''
|
|
|
| annos = bit_2_json(txt)
|
| print(annos)
|
|
|
