Diagramma di dispersione Python con array mascherati con numpy: python, numpy, plot, scatter

Sono bloccato cercando di mascherare i dati per un grafico a dispersione. Tutti i dati sembrano tracciare.

Sto usando array numpy come mostrato nello snippet qui sotto. Sto pensando che forse non posso mascherare sulla matrice "c". Non riesco a trovare alcuna documentazione per farlo, proverò con l'array "s".

Qualsiasi aiuto è molto apprezzato.

yy = NP.ma.array(yy)
xx = NP.ma.array(xx)
zz_masked = NP.ma.masked_where(zz <= 1.0e6 , zz)
scatter(xx,yy,s=15,c=zz_masked, edgecolors="none")
cbar = colorbar()
show()

risposte:

Per me va bene. Ogni chiamata a scatter () ottiene la propria barra di colore poiché ogni colore di scatter () viene normalizzato ai propri dati. Quale versione di matplotlib stai usando?

import pylab as plt
import numpy as np

x = np.linspace(0, 1, 100)
y = x**2
z = y
z_masked = np.ma.masked_where(z > 0.5, z)

plt.scatter(x, y, c=z, s=15, edgecolors="none")
plt.colorbar()
plt.scatter(x+1, y, c=z_masked, s=15, edgecolors="none")
plt.colorbar()
plt.show()