001.3D_Maximal_Independent_Set_Animation.py

import networkx as nx
import plotly.graph_objects as go
from IPython.display import display
import time

# Create a graph using NetworkX
G = nx.Graph()
G.add_edges_from([(1, 2), (1, 3), (2, 4), (3, 4), (4, 5), (4, 6)])

# Convert the graph to a Plotly figure
pos = nx.spring_layout(G)  # Layout for positioning nodes

# Create an empty figure
fig = go.Figure()

# Create frames for animation
frames = []
for i, edge in enumerate(G.edges()):
    edge_trace = go.Scatter(
        x=[pos[edge[0]][0], pos[edge[1]][0], None],
        y=[pos[edge[0]][1], pos[edge[1]][1], None],
        line=dict(width=3, color="rgba(0,0,0,0.2)"),
        hoverinfo="none",
        mode="lines",
        name="Edges",
    )

    node_trace = go.Scatter(
        x=[pos[node][0] for node in G.nodes()],
        y=[pos[node][1] for node in G.nodes()],
        text=[f"Node {node}" for node in G.nodes()],
        mode="markers",
        hoverinfo="text",
        marker=dict(
            showscale=True,
            colorscale="YlGnBu",
            size=20,
            colorbar=dict(tickvals=[], ticktext=[]),
        ),
        name="Nodes",
    )

    # Create a frame for each step
    frame = go.Frame(data=[edge_trace, node_trace], name=f"Step {i + 1}")
    frames.append(frame)

    # Add the initial frame to the figure
    if i == 0:
        fig.add_trace(edge_trace)
        fig.add_trace(node_trace)

# Update layout for the initial frame
fig.update_layout(
    title="Maximal Independent Set Animation",
    titlefont_size=20,
    showlegend=True,
    hovermode="closest",
    margin=dict(b=0, l=0, r=0, t=40),
    legend=dict(
        x=0,
        y=1,
        traceorder="normal",
        font=dict(family="sans-serif", size=12, color="black"),
        bgcolor="LightSteelBlue",
        bordercolor="Black",
        borderwidth=1,
    ),
)

# Add frames to the figure
fig.frames = frames

# Add a play button
play_button = dict(
    type="buttons",
    x=0.92,
    y=0.01,
    xanchor="right",
    yanchor="bottom",
    showactive=False,
    buttons=[
        dict(
            label="Play",
            method="animate",
            args=[
                [f"Step {i + 1}" for i in range(len(frames))],
                {
                    "frame": {"duration": 800, "redraw": True},
                    "fromcurrent": True,
                    "transition": {"duration": 300, "easing": "quadratic-in-out"},
                },
            ],
        )
    ],
)

fig.update_layout(updatemenus=[play_button])

# Display the figure
display(fig)