在现有的matplotlib散点图中添加数据点

Question

如何向现有图表添加数据点？直接的解决方案是绘制一个新的散点图，添加新的数据。

如何向现有图表添加数据点？直接的解决方案是绘制一个新的散点图，添加新的数据。

ax.scatter(data[:,0], data[:,1], cmap = cmap, c = color_data)
ax.scatter(new_points_x, new_points_y, color='blue')

但是，如果我们想要添加新颜色的更多点，就会遇到一个问题：我们必须考虑所有先前添加的点。

如果我可以使用特殊函数，像：

AddPoint(ax, new_point, color)

我只想添加新颜色的新点，不需要任何动画效果。

- toyewonug

---

可能是Dynamically updating plot in matplotlib的重复问题。 - DatHydroGuy

@DatHydroGuy 是的，我以前看过它，但我不需要任何动画等等。我只需要添加一些新数据和新颜色。 - toyewonug

所以第一个散点图有一个颜色映射，而你想添加的点应该有一种不属于原始颜色映射的颜色？ - ImportanceOfBeingErnest

@ImportanceOfBeingErnest 没错。而且我已经有了一个图表。如何在上面添加点？ - toyewonug

Answer 1

如果只是添加新的数据和颜色，可以直接再次调用scatter函数，并指定颜色参数即可将新点添加到图像中：

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(10)
a = np.random.rand(10)
plt.scatter(x, a, c='blue')
b = np.random.rand(10)
plt.scatter(x, b, c='red')
plt.show()

- b-fg

---

在我看来，这段代码和我展示的那个很像，只是变量名不同罢了。 - toyewonug

是的，但我发布了它，所以它显示您可以多次调用它，它们实际上会叠加在一起。不确定您的问题是什么。请编辑您的问题以使其更清晰。 - b-fg

我有一个现有的图表，想在上面添加一个新点。 - toyewonug

仍不清楚，因为我在代码中已经有一个现有的图，并且我正在向其中添加新的数据点。请展示你的完整代码或一个最小可运行的例子以及你期望的输出结果。 - b-fg

如果我能使用像AddPoint这样的特殊函数就太好了。请看问题的第二部分。 - toyewonug

@b-fg 有很多原因可能会让你想要将新点附加到现有曲线上，而不是每个新点都有一个新的散点。你的解决方案在视觉上是有效的，但是之后你就不能做任何事情了，比如检索数据、将两条曲线相加、改变透明度等等...如果一个 y(x) 曲线具有唯一的 gid 并包含其所有数据，则这些事情都很容易。如果你有多个散点，那么它就变得不必要地困难了。 - Guimoute

Answer 2

“不清楚为什么创建第二个scatter（正如@b-fg所建议的）是不可接受的，但您可以编写以下函数：”

def addPoint(scat, new_point, c='k'):
    old_off = scat.get_offsets()
    new_off = np.concatenate([old_off,np.array(new_point, ndmin=2)])
    old_c = scat.get_facecolors()
    new_c = np.concatenate([old_c, np.array(matplotlib.colors.to_rgba(c), ndmin=2)])

    scat.set_offsets(new_off)
    scat.set_facecolors(new_c)

    scat.axes.figure.canvas.draw_idle()

这使您可以向现有的PathCollection添加新点。

示例：

fig, ax = plt.subplots()
scat = ax.scatter([0,1,2],[3,4,5],cmap=matplotlib.cm.spring, c=[0,2,1])
fig.canvas.draw()  # if running all the code in the same cell, this is required for it to work, not sure why
addPoint(scat, [3,6], 'c')
addPoint(scat, [3.1,6.1], 'pink')
addPoint(scat, [3.2,6.2], 'r')
addPoint(scat, [3.3,6.3], 'xkcd:teal')
ax.set_xlim(-1,4)
ax.set_ylim(2,7)

请注意，我提出的函数非常基础，需要根据使用情况进行更智能的调整。重要的是要意识到 PathCollection 中的 facecolors 数组不一定具有与点数相同的元素数量，因此如果尝试同时添加多个点或者原始点都是相同的颜色等情况，颜色可能会出现奇怪的问题...

- Diziet Asahi

---

“为什么我需要draw画布才能让它工作” - 因为只有这样，面颜色才会从彩色地图和规范中实际应用。您可以通过scat.update_scalarmappable()替换该行以避免绘制所有内容。但是，正如您指出的那样，有许多情况下这种方法会失败，因此应谨慎依赖以此方式生成的任何图。 - ImportanceOfBeingErnest

只是想加一个愚蠢的观点...？我的天啊...我以为Python可以在一行代码或更少的代码中完成任何事情..？哇...我几乎可以用C语言做到那样....呼... - user2052717

Answer 3

假设您已经有了一个情节，您可以创建此功能。

def AddPoint(plot, x, y, color):
    plot.scatter(x, y, c=color)
    plot.clf()
    plot.show()

- Ouédraogo Wendé Gûudi Alban