[已解决]字符串索引

XiaoPaiShen

1. source = '长白山上有一品红，一品红在长白山上'
   
2. search_items = ['长白山', '一品红']
   
3. 
   
4. def find_index(source, search_item):
   
5.     result = []
   
6.     start = 0
   
7.     while True:
   
8.         index = source.find(search_item, start)
   
9.         if index == -1:
   
10.             return result
    
11. 
    
12.         start = index + len(search_item)
    
13.         result.append(index)        
    
14.    
    
15. idx = []
    
16. for item in search_items:
    
17.     result = find_index(source, item)
    
18.     idx.extend(result)
    
19. 
    
20. 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)