交互式地改变matplotlib图的alpha值

Question

我已经看了文档，但我似乎不能弄清楚这是否可能-

我有一个数据集，其中x和y值以及离散的z值。多对(x,y)共用相同的z值。我想做的是，当我将鼠标移到具有特定z值的一个点上时，具有相同z值的所有点的alpha变为1 - 即。，如果所有的alpha值最初都是0.5，我只希望具有相同z值的点变为1。

这里有一个最小的工作示例来说明我所说的内容:

#! /usr/bin/env python
import numpy as np
import matplotlib.pyplot as plt
x  = np.random.randn(100)
y  = np.random.randn(100)
z   = np.arange(0, 10, 1)
z   = np.repeat(z, 10)
im = plt.scatter(x, y, c=z, alpha = 0.5)
plt.colorbar(im)
plt.show()

Answer 1

您可能可以使用第二个情节来实现您想要的效果:

import numpy as np
import matplotlib.pyplot as plt
Z = np.zeros(1000, dtype = [("Z", int), ("P", float, 2)])
Z["P"] = np.random.uniform(0.0,1.0,(len(Z),2))
Z["Z"] = np.random.randint(0,50,len(Z))
def on_pick(event):
    z = Z[event.ind[0]]['Z']
    P = Z[np.where(Z["Z"] == z)]["P"]
    selection_plot.set_data(P[:,0],P[:,1])
    plt.draw()

fig = plt.figure(figsize=(10,10), facecolor='white')
fig.canvas.mpl_connect('pick_event', on_pick)
ax = plt.subplot(111, aspect=1)
ax.plot(Z['P'][:,0], Z['P'][:,1], 'o', color='k', alpha=0.1, picker=5)
selection_plot, = ax.plot([],[], 'o', color='black', alpha=1.0, zorder=10)
plt.show()