Quiero hacer un diagrama de dispersión (usando matplotlib) donde los puntos se sombrean de acuerdo con una tercera variable. Me he acercado mucho a esto:
plt.scatter(w, M, c=p, marker='s')donde w y M son los puntos de datos y p es la variable que quiero sombrear con respecto a.
Sin embargo, quiero hacerlo en escala de grises en lugar de color. ¿Alguien puede ayudar?
Respuestas:
No hay necesidad de configurar manualmente los colores. En su lugar, especifique un mapa de colores en escala de grises ...
import numpy as np
import matplotlib.pyplot as plt
# Generate data...
x = np.random.random(10)
y = np.random.random(10)
# Plot...
plt.scatter(x, y, c=y, s=500)
plt.gray()
plt.show()
O, si prefiere una gama más amplia decmap mapas de color, también puede especificar el kwarg scatter. Para usar la versión inversa de cualquiera de estos, simplemente especifique la " _r" versión de cualquiera de ellos. Por ejemplo, en gray_rlugar de gray. Hay varios mapas de color en escala de grises prefabricados (p gray. Ej gist_yarg. binary, Etc.).
import matplotlib.pyplot as plt
import numpy as np
# Generate data...
x = np.random.random(10)
y = np.random.random(10)
plt.scatter(x, y, c=y, s=500, cmap='gray')
plt.show()mpl.cmtambién está disponible directamente como plt.cm.
plt.contour()o plt.contourf(), pero esa es una pregunta diferente
En matplotlib, los colores grises se pueden dar como una cadena de un valor numérico entre 0-1.
Por ejemploc = '0.1'
Luego puede convertir su tercera variable en un valor dentro de este rango y usarla para colorear sus puntos.
En el siguiente ejemplo, utilicé la posición y del punto como el valor que determina el color:
from matplotlib import pyplot as plt
x = [1, 2, 3, 4, 5, 6, 7, 8, 9]
y = [125, 32, 54, 253, 67, 87, 233, 56, 67]
color = [str(item/255.) for item in y]
plt.scatter(x, y, s=500, c=color)
plt.show()
cargumento, con valores alfa en la cuarta columna. Si la tercera variable es Z, con shape=(n,1), entonces colors = numpy.hstack((numpy.zeros_like(z), numpy.zeros_like(z), numpy.ones_like(z), z/z.max()))da un efecto muy agradable (por supuesto, se puede ajustar).
plt.colorbar()hará eso
A veces es posible que necesite trazar el color con precisión según el caso del valor x . Por ejemplo, puede tener un marco de datos con 3 tipos de variables y algunos puntos de datos. Y quieres hacer lo siguiente,
En este caso, es posible que deba escribir en una función corta para asignar los valores de x a los nombres de colores correspondientes como una lista y luego pasar esa lista al plt.scattercomando.
x=['A','B','B','C','A','B']
y=[15,30,25,18,22,13]
# Function to map the colors as a list from the input list of x variables
def pltcolor(lst):
cols=[]
for l in lst:
if l=='A':
cols.append('red')
elif l=='B':
cols.append('blue')
else:
cols.append('green')
return cols
# Create the colors list using the function above
cols=pltcolor(x)
plt.scatter(x=x,y=y,s=500,c=cols) #Pass on the list created by the function here
plt.grid(True)
plt.show()