python如何画多个散点图

原创

mob649e816880fe 2023-07-15 10:57:12 ©著作权

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

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

Python如何画多个散点图

在数据可视化领域，散点图是一种常用的图表形式，用于展示两个变量之间的相关关系。在某些情况下，我们可能需要在同一个图中同时展示多个散点图，以便更好地比较不同组别之间的差异。本文将介绍如何使用Python画多个散点图，并提供一个具体的问题场景。

问题描述

假设我们有一份关于学生考试成绩的数据集，其中包含两个变量：学习时间和考试成绩。我们想要比较三个班级之间学习时间和考试成绩之间的关系，以便了解不同班级的学习情况和考试表现。

解决方案

为了解决这个问题，我们可以使用Python的数据可视化库matplotlib来画多个散点图。下面是具体的步骤和代码示例：

步骤1：导入所需库

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

import matplotlib.pyplot as plt
import numpy as np

步骤2：生成数据

为了示范，我们假设已经有了一份包含三个班级学生的成绩数据集。我们可以使用numpy的random模块生成随机数据来模拟这个数据集。

# 生成随机数据
np.random.seed(0)
num_students = 100
class_labels = ['Class A', 'Class B', 'Class C']
study_time = np.random.randint(1, 6, size=(num_students, len(class_labels)))
exam_scores = np.random.randint(50, 100, size=(num_students, len(class_labels)))

步骤3：画散点图

接下来，我们需要画三个散点图来比较不同班级之间的学习时间和考试成绩之间的关系。我们可以使用matplotlib的scatter函数来画散点图。

# 画散点图
fig, ax = plt.subplots()
colors = ['r', 'g', 'b']

for i in range(len(class_labels)):
    ax.scatter(study_time[:, i], exam_scores[:, i], c=colors[i], label=class_labels[i])

# 设置图表标题和坐标轴标签
ax.set_title('Study Time vs. Exam Scores')
ax.set_xlabel('Study Time')
ax.set_ylabel('Exam Scores')

# 添加图例
ax.legend()

# 显示图表
plt.show()