欢迎访问宙启技术站
智能推送

使用scatter()函数在Python中展示多个数据集的散点图

发布时间:2023-12-25 00:07:49

scatter()函数是Matplotlib库中的一个函数,用于绘制散点图。它可以展示多个数据集之间的关系,进而帮助我们分析数据分布或者寻找数据中的模式。

scatter()函数的语法如下:

plt.scatter(x, y, s=None, c=None, marker=None, cmap=None, alpha=None, ... )

其中,x和y是数据集的x轴和y轴值,s是散点的大小,c是散点的颜色,marker是散点的形状,cmap是颜色图,alpha是透明度等。

下面我们用一个简单的例子来演示如何使用scatter()函数展示多个数据集的散点图:

import numpy as np
import matplotlib.pyplot as plt

# 生成数据
np.random.seed(0)
x1 = np.random.rand(50)
y1 = np.random.rand(50)
x2 = np.random.rand(50)
y2 = np.random.rand(50)

# 设置散点的大小和颜色
s = np.random.randint(10, 100, 50)
c = np.random.rand(50)

# 绘制散点图
plt.scatter(x1, y1, s=s, c=c, marker='o', cmap='viridis', alpha=0.5)
plt.scatter(x2, y2, s=s, c=c, marker='s', cmap='jet', alpha=0.5)

# 添加标题和标签
plt.title('Scatter Plot')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')

# 显示颜色图例
cbar = plt.colorbar()
cbar.set_label('Color')

# 显示图形
plt.show()

在上述例子中,我们生成了两组数据,每组数据包括50个点的x和y坐标,并使用随机数生成散点的大小和颜色。然后,我们使用scatter()函数分别绘制了两组数据的散点图,其中一个使用圆形散点,另一个使用方形散点。我们还为散点图添加了标题、x轴和y轴的标签,并显示了颜色图例。

通过这个例子,我们可以清楚地看到不同数据集之间的散点分布情况,以及散点的大小和颜色的变化。scatter()函数可以在数据分析中帮助我们从视觉上分析数据的关系和特征,进而为决策提供依据。