python使用CnOcr提取图片表格数据保存EXCEL
时间: 2023-07-30 07:09:21 浏览: 494
Python库 | cnocr-1.2.0-py3-none-any.whl
5星 · 资源好评率100%
可以使用以下步骤来实现:
1. 安装CnOcr库:在命令行中输入 `pip install cnocr` 进行安装。
2. 导入所需库:导入CnOcr、Pillow和OpenPyXL库。
```python
import cnocr
from PIL import Image
from openpyxl import Workbook
```
3. 加载CnOcr模型:使用 `cnocr.CnOcr()` 函数加载CnOcr模型。
```python
ocr = cnocr.CnOcr()
```
4. 读取图片和提取文字:使用Pillow库读取图片,并使用CnOcr提取文字。
```python
img = Image.open("table.png")
text = ocr.ocr(img)
```
5. 处理提取到的文字:将提取到的文字进行处理,得到表格的行列数和每个单元格的内容。
```python
rows = []
cols = 0
for line in text:
row = []
for word in line:
row.append(word[1])
rows.append(row)
if len(row) > cols:
cols = len(row)
```
6. 将表格数据保存到Excel中:使用OpenPyXL库将表格数据保存到Excel中。
```python
wb = Workbook()
ws = wb.active
for r in range(len(rows)):
for c in range(len(rows[r])):
ws.cell(row=r+1, column=c+1, value=rows[r][c])
wb.save("table.xlsx")
```
这样,你就可以将图片中的表格数据提取出来,并保存到Excel文件中了。需要注意的是,CnOcr对图片的要求比较高,如果图片质量不好,可能无法正确识别。
阅读全文