如何在matplotlib中更改选定点的颜色？

import matplotlib.pyplot as plt

plt.scatter(x, y)  # x和y分别表示点的横坐标和纵坐标

fig = plt.gcf()  # 获取当前图形对象
ax = plt.gca()  # 获取当前坐标轴对象

selected_point = plt.ginput(1)  # 鼠标交互选择一个点，返回该点的坐标

for scatter in ax.collections:
    contains, _ = scatter.contains(selected_point)  # 判断选择的点是否在散点对象中
    if contains:
        scatter.set_color('red')  # 将选定点的颜色更改为红色

plt.draw()  # 更新图形显示
plt.show()  # 显示图形