欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python 居然可以在 Excel 中画画你知道吗

程序员文章站 2022-06-03 23:50:34
导语:用 python 读取图片的像素值,然后输出到 excel 表格中,最终形成一幅像素画,也就是电子版的十字绣了。基本思路实现这个需求的基本思路是读取这张图片每一个像素的色彩值,然后给 excel...

导语:

用 python 读取图片的像素值,然后输出到 excel 表格中,最终形成一幅像素画,也就是电子版的十字绣了。

基本思路

实现这个需求的基本思路是读取这张图片每一个像素的色彩值,然后给 excel 里的每一个单元格填充上颜色。既然要读取图片,那就需要用到 pillow 库,操作 excel 需要用到 openpyxl 库,先把这两个库安装好。

pip3 install openpyxl
pip3 install pillow

色值转换

从图片读取的像素块色值是 rgb 值,而 openpyxl 向 excel cell 内填充颜色是十六进制色值,因此咱们先写一个 rgb 和十六进制色值转换的一个函数。

def rgb_to_hex(rgb):
    rgb = rgb.split(',')
    color = ''
    for i in rgb:
        num = int(i)
        color += str(hex(num))[-2:].replace('x', '0').upper()
    return color

excel 的单元格默认是长方形,修改为正方形才不会使图片变形

if h == 1:
  _w = cell.column
  _h = cell.col_idx
  # 调整列宽
  worksheet.column_dimensions[_w].width = 1
 
# 调整行高
worksheet.row_dimensions[h].height = 6

这里用到了双重for循环,外层是`width`,里层是`height`,是一列一列的填充颜色,因此判断`if h == 1`,避免多次调整列宽。

图片转换

有了色值转换函数,接下来要做的操作就是逐行读取图片的 rgb 色值,之后将 rgb 色值转换为十六进制色值填充到 excel 的 cell 中即可。

def img2excel(img_path, excel_path):
    img_src = image.open(img_path)
    # 图片宽高
    img_width = img_src.size[0]
    img_height = img_src.size[1]
 
    str_strlist = img_src.load()
    wb = openpyxl.workbook()
    wb.save(excel_path)
    wb = openpyxl.load_workbook(excel_path)
    cell_width, cell_height = 1.0, 1.0
 
    sheet = wb["sheet"]
    for w in range(img_width):
        for h in range(img_height):
            data = str_strlist[w, h]
            color = str(data).replace("(", "").replace(")", "")
            color = rgb_to_hex(color)
            # 设置填充颜色为 color
            fille = patternfill("solid", fgcolor=color)
            sheet.cell(h + 1, w + 1).fill = fille
    for i in range(1, sheet.max_row + 1):
        sheet.row_dimensions[i].height = cell_height
    for i in range(1, sheet.max_column + 1):
        sheet.column_dimensions[get_column_letter(i)].width = cell_width
    wb.save(excel_path)
    img_src.close()

最后再来个入口函数,就大功告成啦~

if __name__ == '__main__':
    img_path = '/users/xyz/documents/tmp/03.png'
    excel_path = '/users/xyz/documents/tmp/3.xlsx'
    img2excel(img_path, excel_path)

惊艳时刻

激动的心,颤抖的手,来看下最终效果咋样。

Python 居然可以在 Excel 中画画你知道吗

怎么样是不是觉得有那么一丝丝韵味呢...

总结

好啦今日代码分享就到这了,喜欢的记得收藏噢~

到此这篇关于python 居然可以在 excel 中画画你知道吗的文章就介绍到这了,更多相关python excel画画内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!