Matplotlib条形图颜色编码修正：解决缺失类别导致的配色错位

Got it, let's sort out that color mismatch issue in your Matplotlib bar chart—here's exactly how to fix it! The core problem right now is that you're cycling through colors based on the visible categories per state, not tying colors to fixed category labels. When some states are missing categories, this throws off the color alignment entirely.

The Solution: Bind Colors to Fixed Category Labels #

Instead of looping through a color list blindly, create a static color mapping dictionary that links each of your 5 categories to a specific color. This way, no matter which state has (or doesn't have) a category, the color will always match the category name.

Step-by-Step Implementation #

1. Define Your Fixed Color Mapping

First, list all 5 categories and assign each a consistent color. You can use Matplotlib's built-in colormap or pick custom hex codes:

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

# Define all 5 categories and their fixed colors
all_categories = ['Category 1', 'Category 2', 'Category 3', 'Category 4', 'Category 5']
category_color_map = {
    'Category 1': '#1f77b4',   # Blue
    'Category 2': '#ff7f0e',   # Orange
    'Category 3': '#2ca02c',   # Green
    'Category 4': '#d62728',   # Red
    'Category 5': '#9467bd'    # Purple
}

2. Plot with Category-Based Color Selection

Let's use sample data that mimics your scenario (some states missing categories) to demonstrate:

# Simulate your data: some states lack certain categories
data = {
    'State': ['CA', 'CA', 'CA', 'NY', 'NY', 'TX', 'TX', 'TX', 'TX'],
    'Category': ['Category 1', 'Category 2', 'Category 4', 'Category 1', 'Category 3', 'Category 1', 'Category 2', 'Category 3', 'Category 5'],
    'Rates': [3.2, 4.5, 2.1, 5.0, 3.8, 2.9, 4.1, 3.5, 2.7]
}
df = pd.DataFrame(data)

fig, ax = plt.subplots(figsize=(10, 6))
bar_width = 0.15
states = df['State'].unique()

# Loop through each state to plot its bars
for idx, state in enumerate(states):
    # Get data for the current state
    state_df = df[df['State'] == state]
    current_cats = state_df['Category'].tolist()
    current_rates = state_df['Rates'].tolist()
    
    # Fetch colors directly from our mapping dictionary
    bar_colors = [category_color_map[cat] for cat in current_cats]
    
    # Calculate x-positions for the bars
    x_pos = np.arange(len(current_cats)) + idx * bar_width
    ax.bar(x_pos, current_rates, width=bar_width, label=state, color=bar_colors)

# Optional: Set x-ticks to show all 5 categories (even missing ones)
all_x_pos = np.arange(len(all_categories)) + bar_width * (len(states) - 1) / 2
ax.set_xticks(all_x_pos)
ax.set_xticklabels(all_categories)

ax.set_ylabel('Rates')
ax.set_title('Rates by State and Category')
ax.legend(title='State')
plt.tight_layout()
plt.show()

3. Alternative: Use Pandas Pivot for Simpler Plotting

If you prefer using Pandas' built-in plotting, you can pivot your data and apply the color mapping directly to columns:

# Pivot the data to get categories as columns
pivot_df = df.pivot(index='State', columns='Category', values='Rates')

# Plot with fixed colors tied to each category column
pivot_df.plot(
    kind='bar',
    color=[category_color_map[col] for col in pivot_df.columns],
    figsize=(10, 6),
    title='Rates by State and Category'
)
plt.ylabel('Rates')
plt.legend(title='Category')
plt.tight_layout()
plt.show()

This will leave gaps (or NaN values) for missing categories, but the colors will still align correctly with the category labels.

Key Takeaway #

By tying each category to a fixed color via a dictionary, you eliminate the risk of color misalignment when states are missing categories. The color will always match the category name, regardless of which states include it.

内容的提问来源于stack exchange，提问作者bernando_vialli