在 matplotlib 上标记不同的色点

Matplotlib是Python中用于数组二维图的惊人可视化库。 Matplotlib 是一个基于 NumPy 数组构建的多平台数据可视化库，旨在与更广泛的 SciPy 堆栈配合使用。

在本文中，任务是根据列表元素的值 x 小于或等于 7 应该用蓝色着色，大于等于 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()

输出：