Dot Matrix Chart

# === IMPORTS ===
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from pywaffle import Waffle

# === USER-EDITABLE PARAMETERS ===
title = "US Population by Generation (2024)\nEach icon = 3 million people"
figsize = (12, 8)

# === EXAMPLE DATASET ===
# Population data in millions
data = {
    'Generation': ['Baby Boomers', 'Gen X', 'Millennials', 'Gen Z', 'Silent/Greatest', 'Gen Alpha'],
    'Population_M': [70, 65, 72, 68, 20, 35],
    'Color': ['#1f77b4', '#2ca02c', '#9467bd', '#ff7f0e', '#7f7f7f', '#e377c2']
}

df = pd.DataFrame(data)
df['Icons'] = (df['Population_M'] / 3).round().astype(int)

# Print summary
print("=== US Population by Generation ===")
print(f"\nTotal Population: {df['Population_M'].sum()}M")
print(f"Total Icons: {df['Icons'].sum()} (each = 3M people)")
print(f"\nBreakdown:")
for _, row in df.iterrows():
    print(f"  {row['Generation']}: {row['Population_M']}M ({row['Icons']} icons)")

# Create waffle data dictionary
waffle_data = {row['Generation']: row['Icons'] for _, row in df.iterrows()}
waffle_colors = {row['Generation']: row['Color'] for _, row in df.iterrows()}

# === CREATE WAFFLE CHART ===
fig = plt.figure(
    FigureClass=Waffle,
    rows=11,
    columns=10,
    values=waffle_data,
    colors=list(waffle_colors.values()),
    title={'label': title, 'fontsize': 16, 'fontweight': 'bold'},
    labels=[f"{k} ({v*3}M)" for k, v in waffle_data.items()],
    legend={
        'loc': 'lower center',
        'bbox_to_anchor': (0.5, -0.15),
        'ncol': 3,
        'fontsize': 10,
        'framealpha': 0
    },
    figsize=figsize,
    block_arranging_style='snake'
)

plt.tight_layout()
plt.show()
# END-OF-CODE

=== US Population by Generation ===

Total Population: 330M
Total Icons: 111 (each = 3M people)

Breakdown:
  Baby Boomers: 70M (23 icons)
  Gen X: 65M (22 icons)
  Millennials: 72M (24 icons)
  Gen Z: 68M (23 icons)
  Silent/Greatest: 20M (7 icons)
  Gen Alpha: 35M (12 icons)