Python基于KMeans提取图像颜色时Matplotlib绘图崩溃求助

一、处理sklearn的n_init参数未来警告 #

• 问题根源：sklearn 1.2及以上版本中，KMeans的n_init默认值从10改为了'auto'，旧代码里显式设置n_init=10会触发未来警告。
• 两种解决方式：
  • 适配新版本规范：把n_init参数改成'auto'，示例代码：

    from sklearn.cluster import KMeans
    kmeans = KMeans(n_clusters=5, n_init='auto', random_state=42)
    

  • 保留旧逻辑并屏蔽警告：如果需要固定初始化次数为10，添加警告过滤代码：

    import warnings
    warnings.filterwarnings("ignore", category=FutureWarning, module="sklearn")
    

二、修复Matplotlib窗口弹出后立即崩溃的问题 #

从以下几个方向排查修复：

• 切换Matplotlib后端
  部分环境默认后端（如TkAgg、QtAgg）存在兼容性问题，尝试更换后端：

  import matplotlib
  # 可选：用非交互式后端生成图片文件，不弹出窗口
  matplotlib.use('Agg')
  # 或者尝试其他交互式后端，需确保系统有对应依赖
  # matplotlib.use('Qt5Agg')
  # matplotlib.use('TkAgg')
  import matplotlib.pyplot as plt
  

• 检查plt.show()的调用
  确保代码末尾有plt.show()，且没有在它之后立即关闭窗口的逻辑。脚本运行时，plt.show()会阻塞程序直到窗口手动关闭，示例：

  plt.pie(percentages, colors=hex_colors, labels=labels)
  plt.title('Dominant Colors')
  plt.show()  # 必须保留，且位置正确
  

• 排查图像数据处理逻辑
  避免因数据异常导致内存溢出：

  • OpenCV读取图像后要转成RGB格式（默认是BGR）：

    import cv2
    img = cv2.imread('your_image.jpg')
    img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
    

  • 扁平化图像数据时确保维度正确：

    pixel_values = img_rgb.reshape((-1, 3))
    pixel_values = np.float32(pixel_values)
    

• 升级依赖版本
  确保各库版本兼容，建议升级到稳定版：

  pip install --upgrade matplotlib scikit-learn opencv-python numpy
  

内容的提问来源于stack exchange，提问作者Mattew