数据分析--word文档自动化

数据分析–word文档自动化

做为数据分析师 经常要生成分析报告 可以使用docx库进行自动化生成报告

新建文档

from docx import Document
from docx.enum.table import WD_TABLE_ALIGNMENT
def new_document():
    """
    创建新文档
    Returns:

    """
    return Document()

添加标题

def add_head(x, head, scale):
    """
    添加标题
    Args:
        x: Document OBJ
        head: 标题内容
        scale: 标题等级

    Returns:

    """
    x.add_heading(head, scale)

添加表格

def add_table(x, row, cols):
    """
    添加表格
    Args:
        x: Document OBJ
        row: 行数
        cols: 列数

    Returns:
        table obj 表格对象
    """
    return x.add_table(rows=row, cols=cols, style='Table Grid')

添加图片

def doc_add_picture(x, png):
    """
    添加图片
    Args:
        x: Document OBJ
        png: str or list, 图片路径

    Returns:

    """
    if isinstance(png, str):
        x.add_picture(png)
    if isinstance(png, list):
        list(map(x.add_picture, png))

保存文档

def save_document(x, s):
    """
    将文档对象保存为文件
    Args:
        x: Document OBJ
        s: 文件路径

    Returns:

    """
    x.save(s)