2023-10-10 python-从一组颜色中找到与指定颜色最接近的颜色-{K-D树}-记录

摘要:

2023-10-10 python-从一组颜色中找到与指定颜色最接近的颜色-{K-D树}-记录

相关文档:

如何在颜色表中找到与当前颜色最接近的颜色？ - 糯米PHP

https://zh.wikipedia.org/wiki/%E6%9C%80%E9%82%BB%E8%BF%91%E6%90%9C%E7%B4%A2

https://zh.wikipedia.org/wiki/K-d%E6%A0%91

K-D树算法:

import numpy as np  
from scipy.spatial import KDTree  
  

"""  
在颜色列表中找到与目标颜色最接近的颜色。  

参数:  
    target_color (tuple): 目标颜色, 如 (255, 0, 0) 表示红色.  
    color_list (list): 所有颜色的列表, 如 [(r1, g1, b1), (r2, g2, b2), ...].  

返回:  
    tuple: 最接近目标颜色的颜色.  
"""  
def find_closest_color(target_color, color_list):  
    # 将RGB颜色转换为浮点数数组  
    colors = np.array(color_list)  
    colors_float = colors.astype(float)  
  
    # 创建k-d树  
    kdtree = KDTree(colors_float)  
  
    # 找到目标颜色的最近邻  
    dist, idx = kdtree.query([target_color])  
  
    # 返回最近邻颜色  
    return color_list[idx[0]]  
  

color_list = [(255, 0, 0), (0, 255, 0), (0, 0, 255), (255, 255, 0), (0, 255, 255)]  
target_color = (255, 255, 0)  

closest_color = find_closest_color(target_color, color_list)  
print("Closest color to target is", closest_color)