파이썬 RGB 숫자로 색깔 이름 프린트하기 (feat. webcolors)

 

파이썬 RGB 숫자로 색깔 이름 프린트하기(feat. webcolors)

RGB를 입력하면 색깔 이름을 프린트하는 코드를 작성한다. 결과는 다음과 같이 나올 수 있도록 한다.

 

 

Input

cname  = RGB(255, 0, 0)

 

Output

Red

 

Input

cname = RGB(254, 1, 1)

 

Output

It is close to Red.

라이브러리

import webcolors
import matplotlib.pyplot as plt

 

Webcolors

Webcolors 라이브러리는 HTML 및 CSS로 정의된 색상 이름 및 색상 값 형식으로 작업하기 위한 라이브러리입니다.

 

Webcolors Documentation

 

Importing Libraries

import webcolors
import matplotlib.pyplot as plt

 

가장 유사한 색깔을 계산하는 함수

def closest_color(rgb):
    differences = {}
    for color_hex, color_name in webcolors.CSS3_HEX_TO_NAMES.items():
        r, g, b = webcolors.hex_to_rgb(color_hex)
        differences[sum([(r - rgb[0]) ** 2,
                         (g - rgb[1]) ** 2,
                         (b - rgb[2]) ** 2])] = color_name
    return differences[min(differences.keys())]

 

정확한 RED color는 RGB가 (255, 0, 0)이다. (254, 0, 0)은 RED가 아니다.

(254, 0, 0) 색깔을 입력했을 때 RED를 반환하기 위해 가장 가장 가까운 색깔을 찾는 함수이다. 

 

예를 들어, 위 함수에 rgb parameter에 (254, 0, 0)를 패스해도 RED를 반환한다.

 

Full Code

import webcolors
import matplotlib.pyplot as plt
 
 
def closest_color(rgb):
    differences = {}
    for color_hex, color_name in webcolors.CSS3_HEX_TO_NAMES.items():
        r, g, b = webcolors.hex_to_rgb(color_hex)
        differences[sum([(r - rgb[0]) ** 2,
                         (g - rgb[1]) ** 2,
                         (b - rgb[2]) ** 2])] = color_name
 
    return differences[min(differences.keys())]
 
 
# RGB 색깔을 지정
color = (113, 241, 224)
 
try:
    cname = webcolors.rgb_to_name(color)
    print(f"The color is exactly {cname}")
 
except ValueError:
    cname = closest_color(color)
    print(f"The color is closest to {cname}")
 
 
plt.imshow([[color]])
plt.show()

 

결과

color  =  ( 113 ,  241 ,  224 )