Cómo anotar diagramas de dispersión de Matplotlib

Un diagrama de dispersión es una forma útil de visualizar la relación entre dos variables.

Afortunadamente, es fácil crear diagramas de dispersión en Matplotlib usando la función matplotlib.pyplot.scatter () .

También es fácil agregar anotaciones o texto a diagramas de dispersión mediante las funciones annotate () y text () . Este tutorial muestra cómo utilizar estas funciones en la práctica.

Crear diagrama de dispersión básico

El siguiente código muestra cómo crear un diagrama de dispersión básico usando Matplotlib:

importar matplotlib.pyplot como plt

#crear datos
x = [3, 6, 8, 12, 14]
y = [4, 9, 14, 12, 9]

#create scatterplot
 plt. dispersión (x, y)

Anotar un solo punto

Podemos usar el siguiente código para agregar una anotación a un solo punto en la gráfica:

importar matplotlib.pyplot como plt

#crear datos
x = [3, 6, 8, 12, 14]
y = [4, 9, 14, 12, 9]

#create scatterplot
 plt. dispersión (x, y)

#add text 'Here' en las coordenadas (x, y) = (6, 9.5) 
plt. texto (6, 9.5, ' Aquí ')

Diagrama de dispersión con anotación en Matplotlib

Anotar varios puntos

Podemos usar el siguiente código para agregar anotaciones a múltiples puntos en la gráfica:

importar matplotlib.pyplot como plt

#crear datos
x = [3, 6, 8, 12, 14]
y = [4, 9, 14, 12, 9]

#create scatterplot
 plt. dispersión (x, y)

# agregar texto a ciertos puntos 
plt. texto (3, 4.5, ' Esto ')
plt. texto (6, 9.5, ' Eso ')
plt. texto (8.2, 14, ' Aquellos ')

Anotar varios puntos en el diagrama de dispersión de Matplotlib

Anotar todos los puntos

Podemos usar el siguiente código para agregar anotaciones a cada punto del gráfico:

importar matplotlib.pyplot como plt

#crear datos
x = [3, 6, 8, 12, 14]
y = [4, 9, 14, 12, 9]
laboratorios = ['A', 'B', 'C', 'D', 'E']

#create scatterplot 
plt. dispersión (x, y)

#use for loop para agregar anotaciones a cada punto en la gráfica  
para i, txt en enumerate (labs):
    plt. anotar (txt, (x [ i ], y [ i ]))

Anotar el diagrama de dispersión de Matplotlib

De forma predeterminada, las anotaciones se colocan directamente encima de los puntos en el diagrama de dispersión y el tamaño de fuente predeterminado es 10.

El siguiente código muestra cómo ajustar ambas configuraciones para que las anotaciones estén ligeramente a la derecha de los puntos y el tamaño de fuente sea un poco más grande:

importar matplotlib.pyplot como plt

#crear datos
x = [3, 6, 8, 12, 14]
y = [4, 9, 14, 12, 9]
laboratorios = ['A', 'B', 'C', 'D', 'E']

#create scatterplot 
plt. dispersión (x, y)

#use for loop para agregar anotaciones a cada punto en la gráfica  
para i, txt en enumerate (labs):
    plt. anotar (txt, (x [ i ] +. 25, y [ i ]), tamaño de fuente = 12)

Matplotlib anota todos los puntos de la trama

Puede encontrar más tutoriales de Matplotlib aquí .

  • https://r-project.org
  • https://www.python.org/
  • https://www.stata.com/

Deja un comentario

La regresión logística es un método estadístico que utilizamos para ajustar un modelo de regresión cuando la variable de respuesta…
statologos comunidad-2

Compartimos información EXCLUSIVA y GRATUITA solo para suscriptores (cursos privados, programas, consejos y mucho más)

You have Successfully Subscribed!