Marca diferentes puntos de color en matplotlib

Matplotlib es una increíble biblioteca de visualización en Python para gráficos 2D de arrays. Matplotlib es una biblioteca de visualización de datos multiplataforma basada en arrays NumPy y diseñada para funcionar con la pila SciPy más amplia.

En este artículo, la tarea es marcar diferentes puntos de color en un gráfico con la condición de que los valores de los elementos de la lista que digan que x es menor o igual a 7 se coloreen en azul y los que sean mayores se coloreen en magenta. La idea es trazar un gráfico con puntos agrupados bajo una condición en diferentes colores, básicamente para agrupar los grupos en un solo color. Para esto, ejecutamos un bucle para todos los valores de x y asignamos un valor de color al valor correspondiente de x. Se hará una lista de colores azul y magenta especificando el color en el i-ésimo índice.

A continuación se muestra la implementación.

import  numpy as np
import matplotlib.pyplot as plt
  
  
# values of x
x = np.array([1, 2, 3, 4, 5,
              6, 7, 8, 9, 10])
  
# values of y
y = np.array([10, 9, 8, 7, 6, 5,
              4, 3, 2, 1])
  
# empty list, will hold color value
# corresponding to x
col =[]
  
for i in range(0, len(x)):
    if x[i]<7:
        col.append('blue')  
    else:
        col.append('magenta') 
  
for i in range(len(x)):
      
    # plotting the corresponding x with y 
    # and respective color
    plt.scatter(x[i], y[i], c = col[i], s = 10,
                linewidth = 0)
      
  
plt.show()

Producción: