Python-是否有函数或公式来查找rgb代码的互补色?

时间:2016-10-25 07:34:45

标签: python colors rgb python-3.5

我试图在Python 3中找到一个好的公式来计算rgb代码的互补色,例如。 a = b的互补。有没有办法做到这一点?

3 个答案:

答案 0 :(得分:7)

这里是如何直接计算RGB颜色的补码。它给出了与使用colorsys的算法相同的结果,如Iva Klass的回答所示,但在我的测试中,它的速度提高了约50%。请注意,它适用于任何RGB方案,RGB组件是整数还是浮点数并不重要(只要每个组件使用相同的范围!)。

函数hilo实现了一个简单的sorting network来对RGB组件进行排序。

# Sum of the min & max of (a, b, c)
def hilo(a, b, c):
    if c < b: b, c = c, b
    if b < a: a, b = b, a
    if c < b: b, c = c, b
    return a + c

def complement(r, g, b):
    k = hilo(r, g, b)
    return tuple(k - u for u in (r, g, b))

这是一个简短的演示,使用PIL / Pillow。

#!/usr/bin/env python3

''' Complement the colours in a RGB image 

    Written by PM 2Ring 2016.10.08
'''

import sys
from PIL import Image

# Sum of the min & max of (a, b, c)
def hilo(a, b, c):
    if c < b: b, c = c, b
    if b < a: a, b = b, a
    if c < b: b, c = c, b
    return a + c

def complement(r, g, b):
    k = hilo(r, g, b)
    return tuple(k - u for u in (r, g, b))

def complement_image(iname, oname):
    print('Loading', iname)
    img = Image.open(iname)
    #img.show()

    size = img.size
    mode = img.mode
    in_data = img.getdata()

    print('Complementing...')
    out_img = Image.new(mode, size)
    out_img.putdata([complement(*rgb) for rgb in in_data])
    out_img.show()
    out_img.save(oname)
    print('Saved to', oname)

def main():
    if len(sys.argv) == 3:
        complement_image(*sys.argv[1:])
    else:
        fmt = 'Complement colours.\nUsage: {} input_image output_image'
        print(fmt.format(sys.argv[0]))

if __name__ == '__main__':
    main()

输入图像

source image

输出图像

output image

这是complement_image的Numpy版本。在我的机器上它处理&#34;眼镜&#34;图像比前一版本快3.7倍。

import numpy as np

def complement_image(iname, oname):
    print('Loading', iname)
    img = Image.open(iname)
    #img.show()

    in_data = np.asarray(img)
    #print(in_data.shape)

    print('Complementing...')
    lo = np.amin(in_data, axis=2, keepdims=True)
    hi = np.amax(in_data, axis=2, keepdims=True)
    out_data = (lo + hi) - in_data

    out_img = Image.fromarray(out_data)
    #out_img.show()
    out_img.save(oname)
    print('Saved to', oname)

答案 1 :(得分:1)

我认为没有现成的解决方案,但标准库中有一个colorsys模块,它可以提供帮助。

我认为您首先需要将RGB转换为HSV or HSL,然后“旋转”色调,并根据需要转换回RGB。例如(我不确定正确旋转):

from colorsys import rgb_to_hsv, hsv_to_rgb

def complementary(r, g, b):
   """returns RGB components of complementary color"""
   hsv = rgb_to_hsv(r, g, b)
   return hsv_to_rgb((hsv[0] + 0.5) % 1, hsv[1], hsv[2])

答案 2 :(得分:0)

r,g,b = [25,25,25]

def get_complementary(color):
    color = color[1:]
    color = int(color, 16)
    comp_color = 0xFFFFFF ^ color
    comp_color = "#%06X" % comp_color
    return comp_color
hex_val = "#%02x%02x%02x" % (r,g,b)
h = get_complementary(hex_val)
print("Complementary color",h)
h = h.lstrip('#')
print('RGB =', tuple(int(h[i:i+2], 16) for i in (0, 2, 4)))

输出:

互补色#E6E6E6

RGB =(230,230,230)