Python中matplotlib是否有获取绘图指定坐标点颜色的函数？

核心结论 #

• matplotlib的pyplot模块不存在直接传入x/y坐标即可提取对应位置色值的内置函数，你写的<extract_color_function>没有直接对应的原生API。

可行替代方案 #

方案1：自主追踪已绘制元素的坐标与颜色映射 #

适合绘图内容是你自己实现、所有绘制元素信息可控的场景，实现简单无误差：

import matplotlib.pyplot as plt

# 提前初始化用于存储坐标-颜色对应关系的列表
point_records = []

# 绘制点的时候同步存储信息
x = int(input("请输入x坐标: "))
y = int(input("请输入y坐标: "))
target_color = "blue"
plt.plot(x, y, color=target_color, marker="o") # 绘制单个点需要指定marker否则不可见
point_records.append({"x": x, "y": y, "color": target_color})

# 自定义查询函数
def extract_point_color(query_x, query_y):
    for p in point_records:
        if p["x"] == query_x and p["y"] == query_y:
            return p["color"]
    # 未匹配到点则返回画布背景色，默认是white
    return "white"

c = extract_point_color(x, y)
if c == "black":
    # 你要执行的逻辑
    pass
else:
    # 你要执行的逻辑
    pass

方案2：画布像素级取色（通用方案） #

适合所有场景，不管是自己绘制的图形还是加载的图片、复杂可视化结果，直接从渲染后的画布像素中提取色值：

import matplotlib.pyplot as plt
import numpy as np
from matplotlib import colors

x = int(input("请输入x坐标: "))
y = int(input("请输入y坐标: "))
plt.plot(x, y, color="blue", marker="o")

# 获取当前画布与坐标轴对象
fig = plt.gcf()
ax = plt.gca()
# 必须先渲染画布，否则没有像素数据
fig.canvas.draw()

# 将数据坐标转换为画布像素坐标
pixel_x, pixel_y = ax.transData.transform((x, y))
# matplotlib像素坐标y轴方向和数组存储方向相反，需要翻转
pixel_y = fig.canvas.get_width_height()[1] - pixel_y

# 读取整张画布的RGBA像素数组
pixel_arr = np.array(fig.canvas.buffer_rgba())
# 提取对应坐标的RGBA值（0-255整数格式）
rgba_int = pixel_arr[int(pixel_y), int(pixel_x)]
# 可转换为matplotlib常用的0-1浮点格式
rgba_float = tuple(rgba_int / 255)

# 可选：转换为十六进制色值或匹配标准颜色名
color_hex = colors.to_hex(rgba_float, keep_alpha=False)
color_name = colors.cnames.get(color_hex, color_hex)

c = color_name
if c == "black":
    # 你要执行的逻辑
    pass
else:
    # 你要执行的逻辑
    pass

注意事项 #

• 如果绘制的点尺寸很小，坐标转换可能出现微小偏差，可以取目标像素周围3*3范围的像素平均值来避免误差
• 如果你没有提前存储绘制元素的信息，也可以遍历ax.lines、ax.collections等对象获取已绘制点的坐标和颜色属性，不用手动维护映射表

内容的提问来源于stack exchange，提问作者23. Sumedha Das