Matplotlib中基于(x,y)组合为(x,w)数据点设置不同颜色的绘图问题

我明白你想要实现的效果——根据x和y的不同组合，给(x,w)的散点设置对应颜色。你之前的代码问题出在每次循环都绘制了整个数据集，而不是单个点，而且只处理了(0,0)这一种情况，所以达不到预期效果。下面给你两种可行的解决方案，从简单到高效：

方法1：修正循环绘制单个点 #

这种方法和你最初的思路一致，但修正了循环中的错误，确保每个点单独绘制并匹配对应颜色：

import matplotlib.pyplot as plt

def plot_points(x, y, w):
    # 先定义(x,y)组合到颜色的映射，替换成你需要的color1/color2等实际颜色值
    color_mapping = {
        (1, 1): "#FF5733",  # 示例颜色：珊瑚红
        (1, 0): "#33FF57",  # 示例颜色：亮绿
        (0, 1): "#3357FF",  # 示例颜色：深蓝
        (0, 0): "#F3FF33"   # 示例颜色：亮黄
    }
    
    # 遍历每个(x,y,w)三元组，绘制单个点
    for xi, yi, wi in zip(x, y, w):
        # 从映射中获取对应颜色，用'o'表示圆点标记
        plt.plot(xi, wi, 'o', color=color_mapping[(xi, yi)])
    
    plt.xlabel('x')
    plt.ylabel('w')
    plt.show()

# 测试示例数据
x = [0,1,0,0,0,1,0]
w = [5,3,3.4,5,3,5,2.5]
y = [1,1,0,1,0,0,0]
plot_points(x, y, w)

方法2：用scatter批量绘制（更高效） #

当数据量较大时，循环绘制单个点效率很低，matplotlib的scatter函数支持直接传入颜色列表，一次性完成绘制，这是更推荐的方式：

import matplotlib.pyplot as plt

def plot_points(x, y, w):
    color_mapping = {
        (1, 1): "#FF5733",
        (1, 0): "#33FF57",
        (0, 1): "#3357FF",
        (0, 0): "#F3FF33"
    }
    
    # 生成每个点对应的颜色列表
    point_colors = [color_mapping[(xi, yi)] for xi, yi in zip(x, y)]
    
    # 批量绘制所有散点，x轴是x列表，y轴是w列表，c参数传入颜色列表
    plt.scatter(x, w, c=point_colors, marker='o')
    
    plt.xlabel('x')
    plt.ylabel('w')
    plt.show()

# 测试调用
x = [0,1,0,0,0,1,0]
w = [5,3,3.4,5,3,5,2.5]
y = [1,1,0,1,0,0,0]
plot_points(x, y, w)

为什么你的原代码没生效？ #

你的原代码中，每次循环执行plt.plot(x,w,'ro')时，都会把整个x和w数组的所有点都画成红色，而不是当前循环的那个点。而且你只处理了(0,0)的情况，其他组合的点根本没有被设置颜色，最后所有点都会被多次覆盖，自然看不到预期的颜色区分效果。

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