import json
import pandas as pd
def generate_category_map(df):
"""
Takes a DataFrame, performs category-based analysis, and generates category_map.html.
"""
# Calculate total address count for each CategoryCode
category_address_counts = df.groupby('CategoryCode')['Address Count'].sum().to_dict()
# Assign a distinct color to each category for the map
unique_categories = sorted(list(df['CategoryCode'].unique()))
colors = [
'#e6194b', '#3cb44b', '#ffe119', '#4363d8', '#f58231', '#911eb4',
'#46f0f0', '#f032e6', '#bcf60c', '#fabebe', '#008080', '#e6beff'
]
category_colors = {category: colors[i % len(colors)] for i, category in enumerate(unique_categories)}
# Prepare data for embedding in HTML's JavaScript
data_for_json = df.to_dict(orient='records')
with open('category_map.html', 'w') as f:
f.write(f'''
Category Map
''')