建议使用批量导入数据,大大缩短时间
for data in open(self.data_path): disease_dict = {} data_json = json.loads(data) disease = data_json['name'] disease_dict['name'] = disease diseases.append(disease) disease_dict['desc'] = '' disease_dict['prevent'] = '' disease_dict['cause'] = '' disease_dict['easy_get'] = '' disease_dict['cure_department'] = '' disease_dict['cure_way'] = '' disease_dict['cure_lasttime'] = '' disease_dict['symptom'] = '' disease_dict['cured_prob'] = ''
if 'symptom' in data_json: symptoms += data_json['symptom'] for symptom in data_json['symptom']: rels_symptom.append([disease, symptom])
if 'acompany' in data_json: for acompany in data_json['acompany']: rels_acompany.append([disease, acompany])
if 'desc' in data_json: disease_dict['desc'] = data_json['desc']
if 'prevent' in data_json: disease_dict['prevent'] = data_json['prevent']
if 'cause' in data_json: disease_dict['cause'] = data_json['cause']
if 'get_prob' in data_json: disease_dict['get_prob'] = data_json['get_prob']
if 'easy_get' in data_json: disease_dict['easy_get'] = data_json['easy_get']
if 'cure_department' in data_json: cure_department = data_json['cure_department'] if len(cure_department) == 1: rels_category.append([disease, cure_department[0]]) if len(cure_department) == 2: big = cure_department[0] small = cure_department[1] rels_department.append([small, big]) rels_category.append([disease, big])
内容来源: liuhuanyong/QASystemOnMedicalKG