一个用于提取简体中文字符串中省,市和区并能够进行映射,检验和简单绘图的python模块。
首先安装cpca库:
pip install cpca
Github:
基本使用
cpca.transform()方法可以输入任意的可迭代类型(如list,pandas的Series类型等),然后将其转换为一个DataFrame。
示例代码:
import cpca
location_str = ["徐汇区虹漕路461号58号楼5楼", "泉州市洛江区万安塘西工业区", "北京朝阳区北苑华贸城"]
df = cpca.transform(location_str)
print(df)
运行结果:
如果想获知程序是从字符串的哪个位置提取出省市区名的,可以添加一个pos_sensitive=True参数
示例代码:
import cpca
location_str = ["徐汇区虹漕路461号58号楼5楼", "泉州市洛江区万安塘西工业区", "北京朝阳区北苑华贸城"]
df = cpca.transform(location_str, pos_sensitive=True)
print(df)
运行结果:
其中省_pos,市_pos和区_pos三列大于-1的部分就代表提取的位置。-1则表明这个字段是靠程序推断出来的,或者没能提取出来。
中国的区级行政单位非常的多,经常有重名的情况,比如“北京市朝阳区”和“吉林省长春市朝阳区”,当有上级地址信息的时候,cpca 能够根据上级地址 推断出这是哪个区,但是如果没有上级地址信息,单纯只有一个区名的时候, cpca 就没法推断了,只能随便选一个了, 通过 umap 参数你可以指定这种情况下该选择哪一个。
示例代码:
import cpca
location_str = ["朝阳区汉庭酒店大山子店"]
df = cpca.transform(location_str, pos_sensitive=True)
print(df)
df = cpca.transform(location_str, pos_sensitive=True, umap={"朝阳区": "110105"})
print(df)
运行结果:
其中:umap 字典的 key 是区名,value 是区的 adcode。具体的 adcode 可以去 全国行政区划查询平台 上查询
转化为列表list
1.转换单一地址
示例代码:
import cpca
location_str = ["北京朝阳区汉庭酒店大山子店"]
ret = cpca.transform(location_str).values.tolist()[0]
print(ret)
运行结果:
2.转换多个地址
示例代码:
import cpca
location_str = "北京朝阳区汉庭酒店大山子店和吉林省朝阳区汉庭酒店大山子店以及北京朝阳区北苑华贸城"
ret = cpca.transform_text_with_addrs(location_str) # 注意 transform_text_with_addrs 获得的数据,“地址”列都是空的
print(ret)
运行结果:
地图绘制
模块中自带一些简单绘图工具,可以在地图上将上面输出的数据以热力图的形式画出来。
这个工具依赖folium,在使用之前需要先安装folium库:
pip install folium
示例代码:
import cpca
from cpca import drawer
location_str = ["徐汇区虹漕路461号58号楼5楼", "泉州市洛江区万安塘西工业区", "北京朝阳区北苑华贸城"]
df = cpca.transform(location_str)
drawer.draw_locations(df[cpca._ADCODE], "df.html")
运行结果:
上述代码运行结束后会在运行代码的当前目录下生成一个df.html文件,用浏览器打开即可看到绘制好的地图(如果某条数据’省’,'市’或’区’字段有缺,则会忽略该条数据不进行绘制),速度会比较慢,需要耐心等待,绘制的图像如下:
关于cpca库更多使用方法,详见上述提到的Github链接。