[已解决]字符串索引

XiaoPaiShen

source = '长白山上有一品红，一品红在长白山上'
search_items = ['长白山', '一品红']

def find_index(source, search_item):
    result = []
    start = 0
    while True:
        index = source.find(search_item, start)
        if index == -1:
            return result

        start = index + len(search_item)
        result.append(index)        
    
idx = []
for item in search_items:
    result = find_index(source, item)
    idx.extend(result)

print(idx)

pythonsrj

contents = ['长白山上有一品红，一品红在长白山上长白山']
mention_list = ["长白山", "一品红"]
label_mention = {}
for content in contents:
    for mention in mention_list:
        kb_id = []
        for j in range(len(content) - len(mention) + 1):
            if content[j: j + len(mention)] == mention:
                kb_id.append(j)
                label_mention[mention] = kb_id
print(label_mention)