python散点图如何加标签

原创

mob64ca12d32849 2023-10-10 15:28:49 ©著作权

文章标签 数据 python Math 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12d32849的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python散点图如何加标签

散点图是数据可视化中常用的一种图表类型，用于展示两个变量之间的关系。在散点图中加入标签可以更加直观地展示数据，并提供更多的信息。本文将通过一个具体的问题，介绍如何在Python中使用matplotlib库绘制散点图并加上标签。

问题描述

假设有一组数据，包含了学生的数学成绩和物理成绩，现在我们想要绘制一个散点图来展示这两个变量之间的关系，并在图中加上每个学生的姓名作为标签。

方案

步骤一：导入必要的库

首先，我们需要导入一些必要的库，包括matplotlib和numpy。

import matplotlib.pyplot as plt
import numpy as np

步骤二：生成数据

为了演示，我们先生成一些随机的数据作为示例。这里我们生成30个学生的成绩数据。

np.random.seed(0)
n = 30
math_scores = np.random.randint(50, 100, n)
physics_scores = np.random.randint(50, 100, n)
names = ['Student {}'.format(i) for i in range(1, n+1)]

步骤三：绘制散点图

接下来，我们使用matplotlib库绘制散点图。

plt.scatter(math_scores, physics_scores)

步骤四：加入标签

为了在散点图中加入标签，我们需要遍历每个点，并在对应的位置上添加标签。

for i, name in enumerate(names):
    plt.annotate(name, (math_scores[i], physics_scores[i]))

步骤五：设置图表标题和坐标轴标签

最后，我们可以添加一些图表标题和坐标轴标签来进一步说明数据。

plt.title('Math Scores vs Physics Scores')
plt.xlabel('Math Scores')
plt.ylabel('Physics Scores')

完整代码示例

下面是完整的代码示例：

import matplotlib.pyplot as plt
import numpy as np

np.random.seed(0)
n = 30
math_scores = np.random.randint(50, 100, n)
physics_scores = np.random.randint(50, 100, n)
names = ['Student {}'.format(i) for i in range(1, n+1)]

plt.scatter(math_scores, physics_scores)

for i, name in enumerate(names):
    plt.annotate(name, (math_scores[i], physics_scores[i]))

plt.title('Math Scores vs Physics Scores')
plt.xlabel('Math Scores')
plt.ylabel('Physics Scores')

plt.show()

运行以上代码，即可得到一张包含标签的散点图。