Python, Matplotlib, Scatter plot, Change color on the clicked point

Question

I have a simple scatter plot with a picker event.
I want to change the color of the data point I click with mouse.
The code I have will change the color of the whole array.
How can I just change one particular point?

import sys
import numpy as np
import matplotlib.pyplot as plt    
testData = np.array([[0,0], [0.1, 0], [0, 0.3], [-0.4, 0], [0, -0.5]])
    fig, ax = plt.subplots()
    sctPlot, = ax.plot(testData[:,0], testData[:,1], "o", picker = 5)
    plt.grid(True)
    plt.axis([-0.5, 0.5, -0.5, 0.5])

def on_pick(event):
    artist = event.artist
    artist.set_color(np.random.random(3))
    print "click!"
    fig.canvas.draw()

fig.canvas.mpl_connect('pick_event', on_pick)

Jason · Accepted Answer · 2015-10-05 15:42:42Z

10

import sys
import numpy as np
import matplotlib.pyplot as plt

testData = np.array([[0,0], [0.1, 0], [0, 0.3], [-0.4, 0], [0, -0.5]])
fig, ax = plt.subplots()
coll = ax.scatter(testData[:,0], testData[:,1], color=["blue"]*len(testData), picker = 5, s=[50]*len(testData))
plt.grid(True)
plt.axis([-0.5, 0.5, -0.5, 0.5])

def on_pick(event):
    print testData[event.ind], "clicked"
    coll._facecolors[event.ind,:] = (1, 0, 0, 1)
    coll._edgecolors[event.ind,:] = (1, 0, 0, 1)
    fig.canvas.draw()

fig.canvas.mpl_connect('pick_event', on_pick)
plt.show()

answered Oct 5, 2015 at 15:42

Jason

2,7333 gold badges16 silver badges22 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

user3376020 Over a year ago

Is it possible to select more than one marker at the same time to retrieve data behind them? Like drawing in rectangle on the scatter plot and do something with all markers inside the rectangle.

eric Over a year ago

user337 there is a lasso functionality: matplotlib.org/3.2.1/gallery/widgets/…

Collectives™ on Stack Overflow

Python, Matplotlib, Scatter plot, Change color on the clicked point

1 Answer 1

2 Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Linked

Related