¿Cómo colorear el diagrama de dispersión por una variable en Matplotlib?

En este artículo, vamos a ver cómo colorear el diagrama de dispersión por variable en Matplotlib . Aquí usaremos los métodos matplotlib.pyplot.scatter() La biblioteca matplotlib se usa para dibujar un diagrama de dispersión. Los diagramas de dispersión se usan ampliamente para representar relaciones entre variables y cómo el cambio en una afecta a la otra.
 

Sintaxis: matplotlib.pyplot.scatter(x_axis_data, y_axis_data, s=Ninguno, c=Ninguno, marker=Ninguno, cmap=Ninguno, vmin=Ninguno, vmax=Ninguno, alpha=Ninguno, linewidths=Ninguno, edgecolors=Ninguno )

Ejemplo 1: Diagrama de dispersión de colores por valores de variables.

En este ejemplo, vamos a ver cómo colorear un diagrama de dispersión con su valor variable. Aquí trazaremos un diagrama de dispersión simple con datos x e y, luego usaremos los atributos c para colorear el punto (puntos variables del diagrama de dispersión). 

import matplotlib.pyplot as plt 
  
x =[5, 7, 8, 7, 2, 17, 2, 9, 
    4, 11, 12, 9, 6] 
  
y =[99, 86, 87, 88, 100, 86, 
    103, 87, 94, 78, 77, 85, 86] 
  
plt.scatter(x, y, c = 'red') 
  
# To show the plot 
plt.show() 

Producción:

Ejemplo 2: punto de diagrama de dispersión de color con valores dependientes.

En este ejemplo, Graficaremos una variable dependiendo de otra variable. A veces necesitamos una visualización basada en precisión, por lo que, en este caso, puede ayudarnos a visualizar los datos que dependen de otra variable. Aquí usaremos tres valores diferentes para cada punto y usaremos un mapa de colores con datos específicos.

import matplotlib.pyplot as plt 
  
x =[5, 7, 8, 7, 3, 4, 5, 6, 7]
  
y =[99, 86, 87, 88, 4, 43, 34, 22, 12]
  
z = [1,2,3,4, 5, 6, 7, 8, 9]
  
plt.scatter(x, y, c = z , cmap = "magma")

Producción: