想法是这样的,根据一个随手输入的乡村地名,匹配出其严格的五级行政地址。例如输入的”无极县东侯坊乡南池阳村助农点“,便要匹配出”河北省-石家庄市-无极县-东侯坊乡-南池阳村“。后面的这个五级的行政地址是已知存在数据库里的。
大概的思路是首先进行分词,如上面的分成”无极县“,”东侯坊乡“,”南池阳村“,然后再匹配看有没有包含这几个词的五级行政地名。
第一部分:分词
1.构建分词的地名字典
要分词首先要有字典。这里我们已经明确都是行政地名,所以构建这样的地名字典还是很简单的。还是上面的例子”河北省-石家庄市-无极县-东侯坊乡-南池阳村“,拆分成”河北省“,”石家庄市“,”无极县“,”东侯坊乡“,”南池阳村“便就可存入地名字典了。这个地名字典属于是静态且可重复利用的,这部分我便就在数据库端处理了。
创建地名字典表namedict
CREATE TABLE `namedict` ( `id` INT(11) NOT NULL AUTO_INCREMENT, `name` VARCHAR(50) NULL DEFAULT NULL COLLATE ‘utf8_bin‘, PRIMARY KEY (`id`) )
之后往表里插入记录
insert into `namedict` (name) select distinct(substring_index(substring_index(village_name,‘-‘,1),‘-‘,-1)) from new_village -- 河北省 union all select distinct(substring_index(substring_index(village_name,‘-‘,2),‘-‘,-1)) from new_village -- 石家庄市 union all select distinct(substring_index(substring_index(village_name,‘-‘,3),‘-‘,-1)) from new_village -- 无极县 union all select distinct(substring_index(substring_index(village_name,‘-‘,4),‘-‘,-1)) from new_village -- 东侯坊乡 union all select distinct(substring_index(substring_index(village_name,‘-‘,5),‘-‘,-1)) from new_village -- 南池阳村
统计了一下地名字典表namedict里共计33万多条的记录。
2.分词判断
有了地名字典就可以进行分词判断了。比如”无极县“,拿这个词到字典里匹配一下是否存在,就知道是否有这个地名了。而”无极县东“在字典里是不存在的,那就可以知道这个地名是不存在的,就不能这样分词了。这里面要注意匹配时判断的效率问题,要避免全扫描,不能每次匹配判断都把33万条记录都扫描一遍,因为一个名字是要经过多次的分词判断才能够得到结果的。也正是这个原因,我把地名字典的所有条目都加载到python的变量里的,这样能够避免多次地和数据库交互。
a.首先先进行hash判断
把33万条记录存放到python的字典变量dict里,那么通过一次hash判断就知道这个地名是否存在。
b.然后再进行范围扫描判断
我们都知道”石家庄市“和”石家庄“是一个意思,但是在dict里面存”石家庄市“而用”石家庄“去匹配是匹配不到的,这里就需要用到范围扫描。
1)把33万个名字再存放到一个list里面,并以一定的规则从小到大排序,如[a1,a2,a3,a4,...,an],且(a1<a2<a3<a4<...an)。
2)if x<a1 or x>an,那就表示x不在list的范围里,也就表示x不在名字字典的范围里,那就不用考虑了。
3)if not (x<a1 or x>an),那么可以先对排序的列表进行分组,每1000元素个作为一个分组。因为是递增排序的,从最右边(最大的)的分组开始,对每个分组的第一个元素(每个分组中最小的元素)进行比较。如果小于等于了这个分组的第一个元素,就表示了这个x在这个分组的范围之内。再以这个分组的第一个元素为开始,向后查询。如果x<=list[i],那么i就是这个范围的起点;如果x>list[j],那么j就是这个范围的终点。这种方式其实就是参考了数据库的索引范围查找算法(B+Tree index range scan),虽说也不算高效,但也比全扫描要好很多了。
3.分词
分词以两个字为起点开始分词,以“无极县东侯坊乡南池阳村助农点”为例:
“无极”是一个词,“无极县”是一个词,“无极县东”不是一个词,那把“无极县”提取出来后面的“东侯坊乡南池阳村助农点”继续。直到拆分成“无极县”,“东侯坊乡”,“南池阳村”。
二.根据分词对五级村名进行匹配
这部分其实也很简单,只要一条SQL语句就能搞定最后的匹配了:
mysql> select village_name from new_village where village_name like ‘%无极县%东侯坊乡%南池阳村%‘; +------------------------------------------------------------+ | village_name | +------------------------------------------------------------+ | 河北省-石家庄市-无极县-东侯坊乡-南池阳村 | +------------------------------------------------------------+ 1 row in set (0.62 sec)
这样的语句是不能走索引的只能全表扫描,不过好在全表也就66万多条记录,总共就几十M的大小,完全可以缓存在数据库的内存里的。
不过即便如此也还是有可优化的空间的。首先把所有的66万个五级行政村名都放到python的list里面,其后可以采取类似数据库全文索引的方法。全文索引的难点在于分词以及stopword的鉴定,而我们这里的条目天然就是已分词好的:”河北省-石家庄市-无极县-东侯坊乡-南池阳村“,也不存在stopword。这里我们需要做的只是记录下每个词所在列表的位置。还是上面那个例子:
fullindex["无极县"]=set([58414, 58415, 58416, 58417, 58418, 58419, 58420, 58421, 58422, 58423,...])
fullindex["东侯坊乡"]=set([58445, 58446, 58447, 58448, 58449, 58450, 58451, 58452, ...])
fullindex["南池阳村"]=set([58460])
再对这几个集合做交集操作得到set([58460]),那list[58460]就是我们要找的“河北省-石家庄市-无极县-东侯坊乡-南池阳村”。(这里得到的是最好的情况有且仅有一个,也有可能会有多个例如只写了个“河北”,那会把河北省所有的村都匹配出来;也有可能一个也没有,那就是地名写的不对,一个也匹配不出来。)
三.运行情况:
1.占用内存
因为对于数据库的操作就是开始的时候加载了两份数据:地名字典(33万)和五级行政村名(66万),大部分的操作都是通过代码实现的,所以最后统计了下比较重要的几个python变量所占用的内存:
存储地名字典的列表:2678096
存储地名字典的dict:12583184
存储五级村名的列表:5429648
存储分词位置的dict:12583184
总共看下来程序运行大概会占用33M左右的内存。
2.花费时间
整个程序运行一次要47秒左右,其中从数据库加载数据的时间比较长,大概要42秒,程序进行分词再对村名进行匹配大概用了5秒左右。如果把程序改成服务端的,那只要在启动的时候加载一次数据即可。要是在分词匹配村名之前能够确认省份的话,那这一动作就能降到1秒左右,就基本能够满足线上操作的要求了。
最后大致的代码如下:
#!/usr/local/bin/python # -*- coding: utf8 -*- ‘‘‘ Created on 2016年6月12日 @author: PaoloLiu ‘‘‘ from mysqlhelper import * import logging,re,sys,itertools,time def logger(): logging.basicConfig(level=logging.DEBUG, format=‘%(asctime)s [%(levelname)s] [%(filename)s] [%(threadName)s] [line:%(lineno)d] %(message)s‘, datefmt=‘%Y-%m-%d %H:%M:%S‘) def get_list_dict(mysqldb): results=mysqldb.executequery("select name from namedict") list_dict=[] for row in results: list_dict.append(row[0].decode("utf8")) list_dict.sort(cmp=None, key=None, reverse=False) return list_dict def get_dic(list_dict): dic={} for item in list_dict: dic[item]=1 return dic def index_range(list,item): # list要求是已排序的 length=len(list) result={} if item<list[0] or item>list[len(list)-1]: result["error"]=True else: for i in range(length/1000,-1,-1): if item>=list[i*1000]: branch=i*1000 # logging.debug("branch="+str(branch)) break for i in range(branch,length): if list[i]>=item: result["begin"]=i result["end"]=i # logging.debug("result[\"begin\"]="+str(result["begin"])) break for i in range(result["begin"],length): if item<list[i]: result["end"]=i # logging.debug("result[\"end\"]="+str(result["end"])) return result def isexist(name,dic,list_dict): pass result=[] if dic.get(name): result.append(name) else: pass range_result=index_range(list_dict,name) if range_result.get("error"): pass else: for i in range(range_result["begin"],range_result["end"]+1): # logging.debug("name="+name) # logging.debug("listname="+list_dict[i][0]) if re.match(name+".*",list_dict[i]): # logging.debug("name="+name) # logging.debug("listname="+list_dict[i][0]) result.append(list_dict[i]) return result def div_word(name,dic,list_dict): divlist=[] while len(name)>=2: match_flag=0 # logging.debug("name:"+name) for i in range(2,len(name)+1): result=[] result=isexist(name[0:i],dic,list_dict) # logging.debug("test name:"+name[0:i]) if len(result)>0: pass match_flag=1 previous=result else: if match_flag==1: # logging.debug("div name1:"+name[0:i-1]) divlist.append(previous) # logging.debug(previous) name=name[i-1:] # logging.debug("new name:"+name) break if i==len(name) and len(result)>0: # logging.debug("div name2:"+name[0:i]) divlist.append(result) name=name[i:] elif i==len(name): name=name[1:] return divlist def get_fullname_list(mysqldb): strsql="select village_name from new_village;" results=mysqldb.executequery(strsql) fullname_list=[] for row in results: fullname_list.append(row[0].decode("utf8")) return fullname_list def get_fullindex(fullname_list): fullindex={} for i in range (0,len(fullname_list)): split_list=fullname_list[i].split("-") for item in split_list: if fullindex.get(item): fullindex[item].add(i) else: fullindex[item]=set() fullindex[item].add(i) return fullindex def main(): begin=time.time() mysqldb=mysqlhelper("dbaadmin","123456","172.16.2.7","3306","spider") mysqldb.connect() list_dict=get_list_dict(mysqldb) dic=get_dic(list_dict) fullname_list=get_fullname_list(mysqldb) fullindex=get_fullindex(fullname_list) load_data_time=round((time.time()-begin),2) logging.info("加载数据耗时"+str(load_data_time)) logging.info("==============================================") name=u"无极县东侯坊乡南池阳村助农点" divlist=div_word(name, dic, list_dict) for x in itertools.product(*divlist): result=fullindex[x[0]] # logging.debug(x[0]) # logging.debug(fullindex[x[0]]) for i in range(1,len(x)): # logging.info(x[i]) # logging.info(fullindex[x[i]]) result=result&fullindex[x[i]] if len(result)>0: for item in result: # logging.debug(item) logging.info("match name:"+fullname_list[item]) match_time=round((time.time()-begin-load_data_time),2) logging.info("名称匹配耗时"+str(match_time)) logging.info("==============================================") logging.info("list_dict getsizeof:"+str(sys.getsizeof(list_dict))) logging.info("dic getsizeof:"+str(sys.getsizeof(dic))) logging.info("fullname_list getsizeof:"+str(sys.getsizeof(fullname_list))) logging.info("fullindex getsizeof:"+str(sys.getsizeof(fullindex))) mysqldb.close() if __name__ == ‘__main__‘: pass logger() main()