如何将ner结果存储在json/数据库中

问题描述

    import nltk
    from itertools import groupby


    def get_continuous_chunks(tagged_sent):
        continuous_chunk = []
        current_chunk = []

        for token, tag in tagged_sent:
            if tag != "O":
                current_chunk.append((token, tag))
            else:
                if current_chunk: # if the current chunk is not empty
                    continuous_chunk.append(current_chunk)
                    current_chunk = []
        # Flush the final current_chunk into the continuous_chunk, if any.
        if current_chunk:
            continuous_chunk.append(current_chunk)
        return continuous_chunk

    ne_tagged_sent = [('Rami', 'PERSON'), ('Eid', 'PERSON'), ('is', 'O'), ('studying', 'O'), ('at', 'O'), ('Stony', 'ORGANIZATION'), ('Brook', 'ORGANIZATION'), ('University', 'ORGANIZATION'), ('in', 'O'), ('NY', 'LOCATION')]

    named_entities = get_continuous_chunks(ne_tagged_sent)
    named_entities = get_continuous_chunks(ne_tagged_sent)
    named_entities_str = [" ".join([token for token, tag in ne]) for ne in named_entities]
    named_entities_str_tag = [(" ".join([token for token, tag in ne]), ne[0][1]) for ne in named_entities]

    def parser(n,string):
        for i in named_entities_str_tag[n]:
            if i==string:
                pass
            else:
                return i


print named_entities_str_tag
print

我从上面的代码中获得了此输出:

I got this output from the above code:

('PERSON'，'Rami Eid')
(组织"，石溪大学")
("LOCATION"，"NY")
("PERSON"，"GuruRaj Bagali")
(组织"，基督大学")

('PERSON ', 'Rami Eid')
('ORGANIZATION', 'Stony Brook University')
('LOCATION ', 'NY')
('PERSON ', 'GuruRaj Bagali')
('ORGANIZATION', 'Christ University')

但是我希望它应该像PERSON WITH ORGANIZATION AND LOCATION这样的地图，我想以json格式存储它.

But I want it should be map like PERSON WITH ORGANIZATION AND LOCATION I want to store it in json format.

推荐答案

您应该将数据格式化为字典，每个条目都对应一个人，例如:

You should format your data as a dictionary, each entry corresponds to a person like:

import json
data = {
        'Rami Eid':{'job': 'engineer', 'location':'NY'},
        'GuruRaj Bagali':{'job': 'professor', 'location': 'NY'}
       }
#Save it in a json file
json.dump(data, open('path/to_your_file', 'w')

这篇关于如何将ner结果存储在json/数据库中的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！