Python matplotlib 绘制散点图详解建议收藏

Python matplotlib 绘制散点图详解

什么是散点图？

散点图是用于观察两个变量之间关系的一种图表，通常用于研究变量之间的相关性。

如何使用Python的matplotlib库绘制散点图

步骤1：导入matplotlib和numpy库

要使用matplotlib绘制散点图，需要导入matplotlib库和numpy库：

import matplotlib.pyplot as plt
import numpy as np

步骤2：生成示例数据集

可以使用numpy库来生成示例数据集，例如，以下代码将生成100个x和y的值，每个值都在0到10之间随机生成：

x = np.random.rand(100)*10
y = np.random.rand(100)*10

步骤3：使用plot函数绘制散点图

使用plot函数可以绘制散点图：

plt.plot(x, y, 'o')
plt.show()

上述代码中，'o'表示绘制散点图，其形状为圆点。可以根据需要替换为其他字符串，例如：'.'（点）、's'（正方形）、'^'（三角形）等等。

步骤4：自定义散点图

使用matplotlib的各种参数可以自定义散点图，以下是一些常用的自定义参数：

• s - 点的大小
• c - 点的颜色
• alpha - 点的透明度
• linewidths - 边界线宽度

colors = np.random.rand(100)
sizes = np.random.rand(100) * 100

plt.scatter(x, y, c=colors, s=sizes, alpha=0.5)
plt.colorbar()
plt.show()

此外，plt.colorbar()函数可以为图表添加一个颜色条。

示例1：绘制一个简单的散点图

以下示例代码将生成一组数据，然后将其绘制成散点图：

import matplotlib.pyplot as plt
import numpy as np

# 生成示例数据集
x = np.array([1,2,3,4,5])
y = np.array([2,4,5,6,7])

# 绘制散点图
plt.plot(x, y, 'o')
plt.show()

在这个示例中，我们使用了numpy库生成了一个x和y的简单数组，然后使用plot函数绘制散点图。

示例2：自定义散点图

以下示例代码将生成一组随机数据，然后将其绘制成自定义散点图：

import matplotlib.pyplot as plt
import numpy as np

# 生成示例数据集
x = np.random.rand(100)
y = np.random.rand(100)
colors = np.random.rand(100)
sizes = np.random.rand(100) * 100

# 自定义散点图
plt.scatter(x, y, c=colors, s=sizes, alpha=0.5)
plt.colorbar()
plt.show()

在这个示例中，我们使用了numpy库生成了100个随机x和y坐标，然后使用scatter函数绘制了一个自定义散点图，包括使用随机颜色和大小。我们还使用plt.colorbar()函数在图表中添加了一个颜色条。

这就是Python matplotlib 绘制散点图的攻略，希望对您有所帮助，谢谢。