【python】使用openpyxl创建excel并写入数据

python 使用openpyxl创建excel并写入数据

# '''
# 实现功能：
#     在excel中，对应的名称后面，插入图片, 文字等
#
# '''
#
import openpyxl,PIL
from PIL import Image
import os

# # 新建excel
file_path = 'D:/1.xlsx'  # 设置路径
if not os.path.exists(file_path):
    wb = openpyxl.Workbook()
    # wb.save(filename=file_path)
    print("新建Excel：" + file_path + "成功")
    ws=wb.active  #获取工作表
    # 添加索引
    fields=["id", "text", "res1", "res2", "res3", "res4", "res4", "label"]
    for field in range(1,len(fields)+1):   # 写入表头
        _=ws.cell(row=1,column=field,value=str(fields[field-1]))
    wb.save(filename=file_path)
else:
    wb = openpyxl.load_workbook(file_path)
    # #获取sheet页
    wbname='Sheet'
    ws=wb.get_sheet_by_name(wbname)  #获取工作表

for i in range(2,11,1):  # 行，从第二行开始，第一行为表头
    # i行设置第一二列，插入数据
    _ = ws.cell(row=i, column=1, value=str(i-1))
    _ = ws.cell(row=i, column=2, value="aaaaa")
    # ----插入图片----
    for j in range(1, 6): #列
        # 图片路径
        img_file_path = os.path.join('D:/1.jpg')   # 设置图片路径
        # #获取图片
        # img = Image(img_file_path)
        # #设置图片的大小
        # img.width, img.height = (110, 110)
        # 获取图片
        img = Image.open(img_file_path)
        # 设置图片的大小
        img = img.resize((110, 110),Image.ANTIALIAS)
        # 设置表格的宽20和高85
        img_column = chr(65 + j + 1)  # 由A（65）列开始，
        ws.column_dimensions[img_column].width = 20
        ws.row_dimensions[i].height = 85
        # 图片插入名称对应单元格
        ws.add_image(img, anchor=img_column + str(i))
wb.save(filename=file_path)  # 保存