在 matplotlib 上标记不同的色点

原文:https://www . geesforgeks . org/mark-different-color-points-on-matplotlib/

Matplotlib 是 Python 中一个惊人的可视化库,用于数组的 2D 图。Matplotlib 是一个多平台数据可视化库,构建在 NumPy 数组上,旨在与更广泛的 SciPy 堆栈一起工作。

在本文中,任务是根据列表中元素的值(比如 x 小于或等于 7)应该用蓝色表示,而那些更大的值应该用洋红色表示的条件,在图形中标记不同的颜色点。这个想法是绘制一个图表,在一个条件下用不同的颜色将点分组,基本上是用一种颜色将聚类分组。为此,我们对 x 的所有值运行一个循环,并为 x 的相应值分配一个颜色值。将由蓝色和洋红色组成一个列表,指定第 I 个索引处的颜色。

下面是实现。

import  numpy as np
import matplotlib.pyplot as plt

# values of x
x = np.array([1, 2, 3, 4, 5,
              6, 7, 8, 9, 10])

# values of y
y = np.array([10, 9, 8, 7, 6, 5,
              4, 3, 2, 1])

# empty list, will hold color value
# corresponding to x
col =[]

for i in range(0, len(x)):
    if x[i]<7:
        col.append('blue')  
    else:
        col.append('magenta') 

for i in range(len(x)):

    # plotting the corresponding x with y 
    # and respective color
    plt.scatter(x[i], y[i], c = col[i], s = 10,
                linewidth = 0)

plt.show()

输出: