自动化神器!Python 批量读取身份证信息写入 Excel

开发 后端 自动化
百度云 SDK 提供了 Python、Java 等多种语言的支持,Python 版的 SDK 安装很简单,使用pip install baidu-aip即可,支持 Python 2.7+ & 3.x 版本。

[[409092]]

今天分享一个实用技能,利用 Python 批量读取身份证信息写入 Excel。

读取

以图片形式的身份证为例,信息读取我们使用百度文字识别OCR来实现,百度接口提供了免费额度,日常使用基本差不多够了,下面来具体看一下如何使用百度文字识别。

SDK 安装

百度云 SDK 提供了 Python、Java 等多种语言的支持,Python 版的 SDK 安装很简单,使用pip install baidu-aip即可,支持 Python 2.7+ & 3.x 版本。

创建应用

创建应用需要一个百度或百度云账号,注册登录地址为:https://login.bce.baidu.com/?redirect=http%3A%2F%2Fcloud.baidu.com%2Fcampaign%2Fcampus-2018%2Findex.html,登录后将鼠标移到登录头像位置,在弹出菜单中点击用户中心,如图所示:

首次进入需选一下相应信息,如图所示:

选完之后点保存即可。

接着将鼠标移到左侧>符号位置,再选人工智能,点击文字识别,如图所示:

点击之后会进到如下所示图中:

现在,我们就可以点击创建应用了,之后进到如下所示图中:

从上图中我们可以看出百度文字识别OCR能够识别的信息类别非常多,也就是说不只是身份证,如果你有其他信息识别的需求也是可以通过它来快速实现的。

这里我们填一下应用名称和应用描述,填完之后点立即创建即可。

创建完成后返回应用列表,如下图所示:

我们需要用到AppID&API Key&Secret Key这三个值,记录一下。

代码实现

代码实现很简单,几行 Python 代码即可搞定,如下所示:

  1. from aip import AipOcr 
  2.  
  3. APP_ID = '自己的APP_ID' 
  4. API_KEY = '自己的API_KEY' 
  5. SECRET_KEY = '自己的SECRET_KEY' 
  6. # 创建客户端对象 
  7. client = AipOcr(APP_ID, API_KEY, SECRET_KEY) 
  8. # 打开并读取文件内容 
  9. fp = open("idcard.jpg""rb").read() 
  10. # res = client.basicGeneral(fp)  # 普通 
  11. res = client.basicAccurate(fp)  # 高精度 

从上述代码中可以看出识别功能分为普通和高精度两种模式,为了识别率更高,我们这里采用高精度模式。

以如下三张我在网上找的假身份证为例:

因为有多张身份证图片,我们需要写一个方法来进行遍历,代码实现如下:

  1. def findAllFile(base): 
  2.     for root, ds, fs in os.walk(base): 
  3.         for f in fs: 
  4.             yield base + f 

通过识别功能获取到的身份证原始信息格式如下:

  1. {'words_result': [{'words''姓名韦小宝'}, {'words''性别男民族汉'}, {'words''出生1654年12月20日'}, {'words''住址北京市东城区景山前街4号'}, {'words''紫禁城敬事房'}, {'words''公民身份证号码11204416541220243X'}], 'log_id': 1411522933129289151, 'words_result_num': 6} 

写入

证件信息的写入使用 Pandas 来实现。这里我们还需要先将获取的原始证件信息进行预处理以便写入 Excel 中,我们将证件的姓名...住址分别存放在数组中,处理代码实现如下:

  1. for tex in res["words_result"]: 
  2.     row = tex["words"
  3.     if "姓名" in row: 
  4.         names.append(row[2:]) 
  5.     elif "性别" in row: 
  6.         genders.append(row[2:3]) 
  7.         nations.append(row[5:]) 
  8.     elif "出生" in row: 
  9.         births.append(row[2:]) 
  10.     elif "住址" in row: 
  11.         addr += row[2:] 
  12.     elif "公民身份证号码" in row: 
  13.         ids.append(row[7:]) 
  14.     else
  15.         addr += row 

之后就可以很方便的将信息直接写入到 Excel 中了,写入代码实现如下:

  1. df = pd.DataFrame({"姓名": names, "性别": genders, "民族": nations, 
  2.                        "出生": births, "住址": address, "身份证号码": ids}) 
  3. df.to_excel('idcards.xlsx'index=False

看一下写入效果:

到此,我们就实现了身份证信息的批量读写功能。

 

责任编辑:武晓燕 来源: Python小二
相关推荐

2020-02-05 15:29:56

计算机网络设备

2018-01-04 11:28:23

2020-12-31 11:55:56

PythonPlaywright微软

2021-09-01 05:11:13

C# 动态链接库

2022-05-07 07:47:51

ChromeAutoma

2009-06-12 11:34:01

2017-05-02 15:10:56

病历智慧城市数据

2023-10-18 13:57:17

2013-09-11 19:10:59

2021-04-17 23:10:59

Python微软Word

2021-02-01 11:03:36

Python开发邮件

2024-02-26 00:00:01

​win32WindowsCOM

2022-03-21 10:09:08

PythonExcel邮件

2020-11-05 12:56:19

Python办公自动化

2013-12-02 12:44:28

Facebook自动化管理

2023-04-27 07:57:52

AutoIt自动化工具

2020-04-21 10:45:47

PythonWordExcel

2024-05-29 11:16:33

PythonExcel

2021-05-06 13:45:13

PythonExce弹窗

2009-03-23 09:53:43

点赞
收藏

51CTO技术栈公众号