Diagrama de dispersión de Matplotlib; color en función de una tercera variable

Question

Quiero hacer un diagrama de dispersión (usando matplotlib) donde los puntos están sombreados según una tercera variable. Tengo muy estrecha con 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 sombra con respecto a.
Sin embargo, quiero hacerlo en escala de grises en lugar de en color. ¿Alguien puede ayudar?

Answer 1

En matplotlib los colores grises se pueden dar como una cadena de un valor numérico entre 0-1.
Por ejemplo c = '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 que utiliza 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() 

Answer 2

No hay necesidad de ajustar manualmente los colores. En su lugar, especificar un mapa de color 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 un wider range of colormaps, también puede especificar el kwarg cmap-scatter. Para usar la versión invertida de cualquiera de estos, simplemente especifique la versión "_r" de cualquiera de ellos. P.ej. gray_r en lugar de gray. Hay varios mapas de color de escala de grises prefabricados (por ejemplo, gray, 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()