[Python知识库]python-doc以stream形式插入图像

引言

• 最近在使用python-docx，想着再生成的docx中插入图像，并居中
• python-docx的官方文档写的不是很清楚，只给出了将现有图像插入到docx中，但是并没有给出如何将ndarray格式数据插入。

解决方案

from docx import Document
from docx.shared import Cm
from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
import cv2
from io import BytesIO

# 这里假装im是从其他地方得来的
im = cv2.imread('a.jpg')
data = cv2.imencode('.jpg', im)[1]
img_bytes = BytesIO(data.tobytes())

doc = Document()
paragraph = doc.add_paragraph()

# 设置段落整体居中，这样图像就会居中
paragraph.alignment = WD_PARAGRAPH_ALIGNMENT.CENTER
run = paragraph.add_run("")
run.add_picture(img_bytes, width=Cm(13.93))

doc.save('test.docx')