python matplotlib散点图

原创

mob649e81586edc 2023-08-25 09:12:44 ©著作权

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

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

Python Matplotlib散点图

引言

Matplotlib是一个用于绘制2D图表和图形的Python库，它是数据可视化的重要工具之一。其中，散点图是一种常用的图表类型，用于展示两个变量之间的关系。本文将介绍如何使用Python的Matplotlib库绘制散点图，并提供相关的代码示例。

准备工作

在开始之前，我们需要确保已经安装了Matplotlib库。可以使用以下命令进行安装：

pip install matplotlib

绘制散点图

绘制散点图可以使用Matplotlib的scatter函数。该函数接受两个参数：x轴上的数据和y轴上的数据。下面是一个简单的示例：

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y = [2, 4, 6, 8, 10]

plt.scatter(x, y)
plt.show()

运行以上代码，将会得到一个简单的散点图。x轴上的数据为[1, 2, 3, 4, 5]，y轴上的数据为[2, 4, 6, 8, 10]。每个点表示一个坐标，其中x和y分别表示该点在x轴和y轴上的位置。

自定义散点图

散点图样式

我们可以通过设置scatter函数的其他参数来自定义散点图的样式。例如，可以设置点的大小、颜色、形状等。下面是一个自定义样式的示例：

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y = [2, 4, 6, 8, 10]
sizes = [30, 60, 90, 120, 150]
colors = ['r', 'g', 'b', 'y', 'm']
shapes = ['o', 's', 'v', '^', 'd']

plt.scatter(x, y, s=sizes, c=colors, marker=shapes)
plt.show()

在上述示例中，我们通过设置s参数来改变点的大小，c参数来改变点的颜色，marker参数来改变点的形状。

添加标签

我们可以通过annotate函数在散点图上添加文字标签。下面是一个添加标签的示例：

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y = [2, 4, 6, 8, 10]
labels = ['A', 'B', 'C', 'D', 'E']

plt.scatter(x, y)

for i, label in enumerate(labels):
    plt.annotate(label, (x[i], y[i]))

plt.show()

在上述示例中，我们通过annotate函数在每个点的位置添加了一个文字标签。

数据分组

有时候，我们需要将散点图中的点按照某种规则进行分组。可以使用不同的颜色或形状来表示不同的组别。下面是一个数据分组的示例：

import matplotlib.pyplot as plt

x1 = [1, 2, 3, 4, 5]
y1 = [2, 4, 6, 8, 10]
x2 = [1, 2, 3, 4, 5]
y2 = [1, 3, 5, 7, 9]

plt.scatter(x1, y1, c='r', marker='o', label='Group 1')
plt.scatter(x2, y2, c='b', marker='s', label='Group 2')

plt.legend()
plt.show()

在上述示例中，我们将数据分为两个组，分别用红色圆圈和蓝色正方形表示，并使用label参数添加了图例。