Gráfico de burbujas 3D usando Plotly en Python

Plotly es una biblioteca de Python que se utiliza para diseñar gráficos, especialmente gráficos interactivos. Puede trazar varios gráficos y tablas como histograma, diagrama de barras, diagrama de caja, diagrama de dispersión y muchos más. Se utiliza principalmente en el análisis de datos, así como en el análisis financiero. plotly es una biblioteca de visualización interactiva.

Gráfico de burbujas 3D en Plotly

Se puede crear un gráfico de burbujas utilizando el método scatter_3d() de la clase plotly.express. Un gráfico de burbujas es una visualización de datos que ayuda a mostrar múltiples círculos (burbujas) en un gráfico dimensional 3D como lo mismo en un gráfico de dispersión 3D. Un gráfico de burbujas se utiliza principalmente para representar y mostrar relaciones entre variables numéricas.

Ejemplo 1: Uso del conjunto de datos de Iris

import plotly.express as px
  
  
df = px.data.iris()
fig = px.scatter_3d(df, x='sepal_width',
                    y='sepal_length', 
                    z='petal_width',
                    size='petal_length', 
                    color='species')
  
fig.show()

Producción:

Ejemplo 2: uso del conjunto de datos de sugerencias

import plotly.express as px
  
  
df = px.data.tips()
fig = px.scatter_3d(df, x='total_bill',
                    y='day', z='time',
                    size='tip', color='sex')
  
fig.show()

Producción:

Personalización de la barra de colores

Se puede editar usando la opción de color del método Scatter3D() de la clase graph_objects. Veamos el siguiente ejemplo para una mejor comprensión del tema.

Ejemplo:

import plotly.express as px
import plotly.graph_objects as go
  
  
df = px.data.tips()
  
fig = go.Figure(go.Scatter3d(
    x = df['total_bill'],
    y = df['time'],
    z = df['day'],
    mode = 'markers',
    marker = dict(
        color = df['tip'],
        size = df['total_bill'],
        colorscale=[[0, 'rgb(15, 10, 172)'], 
                    [.3, 'rgb(150, 255, 255)'], 
                    [1, 'rgb(100, 10, 100)']]
        )
))
  
fig.show()

Producción: