Добавьте точки к существующему графику рассеяния 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)

. Я хочу добавлять только новые точки в новых цветах.Мне НЕ нужны анимации

Answer 1

Непонятно, почему создание второго 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)

Обратите внимание, что предлагаемая мной функцияочень простой и должен быть сделан намного умнее в зависимости от варианта использования.Важно понимать, что массив facecolors в PathCollection не обязательно имеет то же количество элементов, что и количество точек, поэтому с цветами могут происходить забавные вещи, если вы попытаетесь добавить несколько точек как один раз, или еслиисходные точки имеют одинаковые цвета и т. д.

Answer 2

Если у вас уже есть график, вы можете создать эту функцию.

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

Answer 3

Чтобы просто добавить новые данные с новым цветом, действительно при повторном вызове 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()