Create Sankey Diagrams Instantly

Sample code / prompt

import plotly.graph_objects as go

# US Energy Flow Data (Quadrillion BTU)
sources = ['Coal', 'Natural Gas', 'Petroleum', 'Nuclear', 'Renewables']
source_values = [11, 32, 35, 8, 12]

transforms = ['Electricity Gen.', 'Direct Use', 'Rejected Energy']
end_uses = ['Residential', 'Commercial', 'Industrial', 'Transportation']

# Define flows: source -> transform/enduse

Sample code / prompt

import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd
import numpy as np

# Create correlation matrix for financial metrics
metrics = ['Revenue', 'Profit', 'Expenses', 'ROI', 'Customers', 'AOV', 'Marketing', 'Employees']
correlation_data = np.array([
    [1.00, 0.85, -0.45, 0.72, 0.88, 0.65, 0.72, 0.55],
    [0.85, 1.00, -0.78, 0.92, 0.75, 0.58, 0.63, 0.48],

Sample code / prompt

import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd
import numpy as np
from scipy.stats import f_oneway

# Generate exam score data for 3 groups
np.random.seed(42)
control = np.random.normal(72, 12, 50)
treatment_a = np.random.normal(78, 10, 50)