使用python生成excel格式的图片

使用Python生成Excel格式的图片

分类： Python 2013-07-20 18:42 100人阅读 评论(0) 收藏 举报

Python

之前曾看到过 一篇报道，说有个日本人利用Excel作画。他把每个单元格填充上颜色，缩小显示比例，就展现出一幅画。记者采访他，他说因为买不起正版的Photoshop，才委曲求全用Excel将就。

当时看着几百乘几百的单元格里，填满眼花缭乱的的颜色，心想这得花多大的精力啊？

后来想想，要是用编程来实现，普通人几分钟之内也可以完成几幅，甚至可以达到照片级，呵呵~

先看个效果图，Excel格式的图片，够Geek吧。

用Python来实现，是很简单的，我的平台是:
Python 2.7.5;
PIL 1.1.7;
XlsxWriter 0.3.5;

需要说明的是，之所以选择XlsxWriter，而不是常用的xlwt，是因为前者可以操作Excel2007版本的xlsx文件。它拥有更多的行和列。

主要思路，就是使用PIL打开图片文件，读出每个像素点的RGB值，再填充Excel文件。

需要注意的是，Excel2007对于单元格样式的种类也是有要求的，貌似最多65536种。因此有可能需要对颜色进行圆整。我也曾想使用PIL预先将图片的颜色转为65536色（16 bit），但是貌似PIL对65536色不太支持，所以只能出此下策。

[python] view plain copy print ?

1. # coding: utf-8
3. from PIL import Image
4. from xlsxwriter.workbook import Workbook
6. class ExcelPicture(object):
7. FORMAT_CONSTRAINT = 65536
9. def __init__(self, pic_file, ratio = 1.0):
10. self.__pic_file = pic_file
12. self.__ratio = ratio
13. self.__zoomed_out = False
15. self.__formats = dict()
18. # 缩小图片
19. def zoom_out(self, _img):
20. _size = _img.size
21. _img.thumbnail((int(_img.size[0] * self.__ratio), int(_img.size[1] * self.__ratio)))
23. self.__zoomed_out = True
26. # 对颜色进行圆整
27. def round_rgb(self, rgb, model):
28. return tuple([int(round(x / model) * model) for x in rgb])
31. # 查找颜色样式，去重
32. def get_format(self, color):
33. _format = self.__formats.get(color, None)
35. if _format is None:
36. _format = self.__wb.add_format({'bg_color': color})
37. self.__formats[color] = _format
39. return _format
42. # 操作流程
43. def process(self, output_file = '_pic.xlsx', color_rounding = False, color_rounding_model = 5.0):
44. # 创建xlsx文件，并调整行列属性
45. self.__wb = Workbook(output_file)
46. self.__sht = self.__wb.add_worksheet()
47. self.__sht.set_default_row(height = 9)
48. self.__sht.set_column(0, 5000, width = 1)
50. # 打开需要进行转换的图片
51. _img = Image.open(self.__pic_file)
52. print 'Picture filename:', self.__pic_file
54. # 判断是否需要缩小图片尺寸
55. if self.__ratio < 1: self.zoom_out(_img)
57. # 遍历每一个像素点，并填充对应的颜色到对应的Excel单元格
58. _size = _img.size
59. print 'Picture size:', _size
60. for (x, y) in [(x, y) for x in xrange(_size[0]) for y in xrange(_size[1])]:
61. _clr = _img.getpixel((x, y))
63. # 如果颜色种类过多，则需要将颜色圆整到近似的颜色上，以减少颜色种类
64. if color_rounding: _clr = self.round_rgb(_clr, color_rounding_model)
66. _color = '#%02X%02X%02X' % _clr
67. self.__sht.write(y, x, '', self.get_format(_color))
69. self.__wb.close()
71. # 检查颜色样式种类是否超出限制，Excel2007对样式数量有最大限制
72. format_size = len(self.__formats.keys())
73. if format_size >= ExcelPicture.FORMAT_CONSTRAINT:
74. print 'Failed! Color size overflow: %s.' % format_size
75. else:
76. print 'Success!'
77. print 'Color: %s' % format_size
78. print 'Color_rounding:', color_rounding
79. if color_rounding: print 'Color_rounding_model:', color_rounding_model
82. if __name__ == '__main__':
83. r = ExcelPicture('b022_c.jpg', ratio = 0.5)
84. r.process('b022_c.xlsx', color_rounding = True, color_rounding_model = 5.0)