import os
import pandas as pd
import docx
from docx.shared import Inches
CURRENT_DIR = os.path.abspath(os.path.dirname(__file__))
def save_df_to_doc(document, test_df):
'''
将结果按照dataframe的形式存入doc文件
:param document: 存入的文档类
:param test_df: 需要保存的df
:return:
'''
# add_paragraph表示添加一个段落
document.add_paragraph(u'各个指标的平均值表格\n')
# 添加一个表格--行数和列数,行数多加一行,需要将列名同时保存
t = document.add_table(test_df.shape[0] + 1, test_df.shape[1])
# 将每列列名保存到表格中
for j in range(test_df.shape[-1]):
t.cell(0, j).text = test_df.columns[j]
# 将每列数据保存到新建的表格中
for i in range(test_df.shape[0]):
for j in range(test_df.shape[-1]):
# 第一行保存的是列名,所以数据保存时,行数要加1
t.cell(i + 1, j).text = str(test_df.values[i, j])