以交互方式更改matplotlib图的alpha值

时间:2014-08-06 07:12:02

标签: python matplotlib

我查看了文档,但我似乎无法弄清楚这是否可行 -

我有一个数据集,其中包含xy值以及离散的z值。多对(x,y)共享相同的z值。我想要做的是当我鼠标移动一个具有特定z值的点时,具有相同alpha值的所有点的z变为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()

1 个答案:

答案 0 :(得分:0)

您可以使用第二个图表伪装您想要实现的目标:

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()