使用Matplotlib标记散点图点|极客教程

使用Matplotlib标记散点图点

Matplotlib是一个用于绘制数据可视化图表的Python库。在数据可视化中，有时候我们需要对散点图的点进行标记，以区分不同的数据点。在本文中，我们将学习如何使用Matplotlib对散点图的点进行标记。

首先，我们需要导入Matplotlib库：

import matplotlib.pyplot as plt

1. 使用不同颜色标记不同类别的点

有时候我们需要将散点图中不同类别的点用不同的颜色进行标记。下面是一个示例代码：

import matplotlib.pyplot as plt

# 生成数据
x1 = [1, 2, 3, 4, 5]
y1 = [2, 3, 4, 5, 6]
x2 = [2, 3, 4, 5, 6]
y2 = [1, 2, 3, 4, 5]

# 绘制散点图并标记不同类别的点
plt.scatter(x1, y1, color='blue', label='Class 1')
plt.scatter(x2, y2, color='red', label='Class 2')
plt.legend()
plt.show()

Output:

使用Matplotlib标记散点图点

在上面的示例代码中，我们生成了两个类别的数据点，并将它们用不同的颜色进行标记。

2. 使用不同形状标记不同类别的点

除了使用不同的颜色进行标记外，我们还可以使用不同的形状来标记不同类别的点。下面是一个示例代码：

import matplotlib.pyplot as plt

# 生成数据
x1 = [1, 2, 3, 4, 5]
y1 = [2, 3, 4, 5, 6]
x2 = [2, 3, 4, 5, 6]
y2 = [1, 2, 3, 4, 5]

# 绘制散点图并标记不同形状的点
plt.scatter(x1, y1, marker='s', label='Class 1')  # 使用正方形标记
plt.scatter(x2, y2, marker='^', label='Class 2')  # 使用三角形标记
plt.legend()
plt.show()

Output:

使用Matplotlib标记散点图点

在上面的示例代码中，我们使用了不同的形状来标记不同类别的数据点。

3. 使用不同大小标记不同类别的点

除了颜色和形状外，我们还可以使用不同的大小来标记不同类别的点。下面是一个示例代码：

import matplotlib.pyplot as plt

# 生成数据
x1 = [1, 2, 3, 4, 5]
y1 = [2, 3, 4, 5, 6]
x2 = [2, 3, 4, 5, 6]
y2 = [1, 2, 3, 4, 5]

# 绘制散点图并标记不同大小的点
plt.scatter(x1, y1, s=100, label='Class 1')  # 大小为100
plt.scatter(x2, y2, s=200, label='Class 2')  # 大小为200
plt.legend()
plt.show()

Output:

使用Matplotlib标记散点图点

在上面的示例代码中，我们使用了不同的大小来标记不同类别的数据点。

4. 使用文本标记数据点

有时候我们需要在散点图中用文本标记数据点，以显示数据的具体数值。下面是一个示例代码：

import matplotlib.pyplot as plt

# 生成数据
x = [1, 2, 3, 4, 5]
y = [2, 3, 4, 5, 6]

# 绘制散点图并用文本标记数据点
for i in range(len(x)):
    plt.scatter(x[i], y[i])
    plt.text(x[i], y[i], f'Point {i+1}', fontsize=12)

plt.show()