Marimekko Chart

# === IMPORTS ===
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle

# === USER-EDITABLE PARAMETERS ===
title = "Smartphone Market Share by Region ($550B Total)"
figsize = (14, 8)

# === EXAMPLE DATASET ===
# Regions and their market sizes (in billions)
regions = ['North America', 'Europe', 'Asia-Pacific', 'Rest of World']
market_sizes = [150, 120, 200, 80]

# Companies and their shares in each region
companies = ['Apple', 'Samsung', 'Xiaomi', 'Others']
company_colors = {'Apple': '#555555', 'Samsung': '#1428A0', 'Xiaomi': '#FF6900', 'Others': '#CCCCCC'}

# Market share percentages by region
shares = {
    'North America': [55, 25, 5, 15],
    'Europe': [35, 40, 15, 10],
    'Asia-Pacific': [20, 25, 35, 20],
    'Rest of World': [15, 30, 30, 25]
}

# Print summary
print("=== Smartphone Market Share ===")
print(f"\nTotal Market: ${sum(market_sizes)}B")
print(f"\nBy Region:")
for reg, size in zip(regions, market_sizes):
    print(f"  {reg}: ${size}B ({size/sum(market_sizes)*100:.0f}%)")

print(f"\nCompany shares by region:")
for reg in regions:
    print(f"\n  {reg}:")
    for comp, share in zip(companies, shares[reg]):
        print(f"    {comp}: {share}%")

# === CREATE MARIMEKKO CHART ===
fig, ax = plt.subplots(figsize=figsize)

# Normalize widths
total_market = sum(market_sizes)
widths = [size / total_market for size in market_sizes]

# Draw rectangles
x_pos = 0
for i, (region, width) in enumerate(zip(regions, widths)):
    y_pos = 0
    for j, (company, share) in enumerate(zip(companies, shares[region])):
        height = share / 100
        rect = Rectangle(
            (x_pos, y_pos), width, height,
            facecolor=company_colors[company],
            edgecolor='white',
            linewidth=2
        )
        ax.add_patch(rect)
        
        # Add label if segment is large enough
        if share >= 10:
            ax.text(
                x_pos + width/2, y_pos + height/2,
                f'{company}\n{share}%',
                ha='center', va='center',
                fontsize=9, fontweight='bold',
                color='white'
            )
        
        y_pos += height
    
    # Region label at bottom
    ax.text(x_pos + width/2, -0.05, f'{region}\n${market_sizes[i]}B',
            ha='center', va='top', fontsize=10, fontweight='bold')
    
    x_pos += width

# Styling
ax.set_xlim(0, 1)
ax.set_ylim(-0.15, 1)
ax.set_aspect('equal')
ax.axis('off')

# Legend
from matplotlib.patches import Patch
legend_elements = [Patch(facecolor=color, label=comp) 
                   for comp, color in company_colors.items()]
ax.legend(handles=legend_elements, loc='upper left', framealpha=0.9)

ax.set_title(title, fontsize=16, fontweight='bold', pad=20)

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

=== Smartphone Market Share ===

Total Market: $550B

By Region:
  North America: $150B (27%)
  Europe: $120B (22%)
  Asia-Pacific: $200B (36%)
  Rest of World: $80B (15%)

Company shares by region:

  North America:
    Apple: 55%
    Samsung: 25%
    Xiaomi: 5%
    Others: 15%

  Europe:
    Apple: 35%
    Samsung: 40%
    Xiaomi: 15%
    Others: 10%

  Asia-Pacific:
    Apple: 20%
    Samsung: 25%
    Xiaomi: 35%
    Others: 20%

  Rest of World:
    Apple: 15%
    Samsung: 30%
    Xiaomi: 30%
    Others: 25%