# -*- coding: UTF-8 -*-

import Image
import math

beer = Image.open('beer2.png')
#获取颜色信息
colors = beer.getcolors()
colors_len = len(colors)
data = list(beer.getdata())
colors_index = colors_len - 1
while colors_index > 0:
    #创建一幅新的图片，以像素数总数的开平方数为长宽值
    dim = math.sqrt(len(data))
    new_img = Image.new('1', (int(dim), int(dim)))
    #把所有以最大颜色值为颜色参数的像素点标记为‘亮’，其余像素点标记为‘暗’
    output = [item == colors[colors_index][1] for item in data]
    new_img.putdata(output)
    new_img.save('level33_%d.png'%(colors_index / 2))
    #抛弃颜色值最大的一组颜色，并且抛弃使用这些颜色的像素
    eli_color_a = colors.pop(colors_index)[1]
    eli_color_b = colors.pop(colors_index - 1)[1]
    for k in xrange(len(data) - 1, -1, -1):
        if data[k] == eli_color_a or data[k] == eli_color_b:
            data.pop(k)
    colors_index -= 2