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í ')

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 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 ]))

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)

Puede encontrar más tutoriales de Matplotlib aquí .

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