¿Cómo hacer un gráfico de burbujas con Altair en Python?

Requisito previo: Introducción a Altair en Python

Altair es una biblioteca de visualización estadística simple y fácil de usar para Python. Contiene muchos tipos de gráficos incorporados y varias opciones para modificar las propiedades y generar otros gráficos. Bubble Plot es una visualización muy útil para el análisis bivariado de datos con respecto a una tercera variable. No está disponible en la biblioteca de Altair, pero se puede hacer haciendo algunas modificaciones simples al diagrama de dispersión.

¿Qué es un gráfico de burbujas?

Bubble Plot es básicamente un gráfico de dispersión entre dos variables/columnas de datos donde, en lugar de los puntos de datos, hay burbujas/círculos de diferentes tamaños que indican la tercera variable. La tercera variable puede ser de tipo cuantitativo, ordinal o nominal, pero el mejor tipo para usar en el gráfico de burbujas es el tipo ordinal, es decir, datos que tienen un orden específico. La leyenda muestra qué tamaño de círculo corresponde a qué valor de datos.

Un gráfico de burbujas puede ayudarnos a ver la relación entre dos variables con respecto a una tercera variable. Cuanto más grande es la burbuja, mayor es el valor de los datos a los que corresponde.

Creación de un gráfico de burbujas

Para hacer un gráfico de burbujas, el usuario simplemente tiene que asignar una variable adecuada del conjunto de datos a la codificación de tamaño en un gráfico de dispersión simple.

Los conjuntos de datos utilizados en estos artículos son de la biblioteca Vega_datasets. 

# Python3 program to illustrate 
# How to make a bubble plot
# using the altair library
    
# Importing altair and vega_datasets 
import altair as alt 
from vega_datasets import data 
    
# Selecting the cars dataset 
cars = data.cars() 
    
# Making the base scatter plot 
alt.Chart(cars).mark_point().encode( 
    
  # Map the sepalLength to x-axis 
    x = 'Acceleration', 
    
  # Map the petalLength to y-axis 
    y = 'Displacement',
    
  # Map the Cylinders variable to size
  # and specify it as a nominal variable
      size = 'Cylinders:N'
)

Producción:

Diagrama de burbujas simple usando Altair

Personalización del gráfico de burbujas

Puede realizar las siguientes personalizaciones en el gráfico de burbujas:

• Color : puede cambiar el color predeterminado de las burbujas configurando el parámetro de color del método mark_point().
• Opacidad : puede cambiar la opacidad predeterminada de las burbujas configurando el parámetro de opacidad del método mark_point(). Va de 0 a 1.
• Relleno : esto es falso por defecto, pero puede cambiar el parámetro relleno a verdadero, llenando así la burbuja con el color especificado.

Ejemplo:

# Python3 program to illustrate
# how to customize a bubble plot
  
# Importing altair and vega_datasets
import altair as alt
from vega_datasets import data
  
# Selecting the cars dataset
cars = data.cars()
  
# Making the base scatter plot
# and adding the customizations
alt.Chart(cars).mark_point(color='green',
                           filled=True,
                           opacity=0.4).encode(
    
    # Map the sepalLength to x-axis
    x='Acceleration',
    
    # Map the petalLength to y-axis
    y='Displacement',
    
    # Map the Cylinders variable to size
    # and specify it as a nominal variable
    size='Cylinders:N'
)

Producción:

Gráfico de burbujas personalizado con Altair