如何在Python中绘制z值对应指定色键的散点图？

Hey there! To get your scatter plot points colored correctly based on the z array and your custom colors list, you just need to map each value in z to its corresponding color from the list. Here's how to do it step by step:

Step 1: Understand the Mapping #

Your colors list is indexed from 0 to 8, and each z value directly corresponds to one of these indices. For example, z=4 uses colors[4] = '#9C6BE1' as you specified.

Step 2: Generate the Color List for Each Point #

Create a new list where each element is the color from colors that matches the corresponding value in z. You can do this easily with a list comprehension.

Step 3: Plot with the Custom Colors #

Pass this generated color list to the c parameter of plt.scatter().

Full Working Code Example #

import matplotlib.pyplot as plt
from matplotlib.lines import Line2D

# Your input data
colors = ["#049FBB", "#5DF10A", "#BF5B3F", "#A95058", "#9C6BE1","#6CC43C","#7B3001", "#B3D407","#97CE66"]
x = [1,2,3,4]
y = [3,4,8,6]
z = [4,6,8,2]

# Map each z value to its corresponding color
point_colors = [colors[val] for val in z]

# Create the scatter plot
plt.scatter(x, y, c=point_colors, s=100)  # s increases point size for visibility

# Optional: Add a legend to show z-value ↔ color mapping
legend_handles = []
for z_val, color in enumerate(colors):
    # Create a custom marker handle for each z value
    handle = Line2D(
        [0], [0], marker='o', color='w', 
        markerfacecolor=color, markersize=10, 
        label=f'z={z_val}'
    )
    legend_handles.append(handle)

plt.legend(handles=legend_handles, bbox_to_anchor=(1.05, 1), loc='upper left')
plt.xlabel('X Axis')
plt.ylabel('Y Axis')
plt.title('Scatter Plot with Custom Z-Value Color Mapping')
plt.tight_layout()  # Adjust layout to fit the legend
plt.show()

Key Notes: #

• The list comprehension [colors[val] for val in z] is the core of the solution—it takes each value in z and uses it as an index to fetch the right color from your colors list.
• If your z values ever go outside the range 0-8 (the length of colors minus 1), you'll want to add error handling (like checking if val is a valid index) to avoid index errors.
• The legend part is optional but helpful for readers to understand which color corresponds to which z value. We use Line2D to create custom marker handles since matplotlib doesn't auto-generate a legend for direct color lists.

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