python画散点图四种不同数值的数据不同颜色

要实现四种不同数值的数据点显示不同颜色的散点图，可以使用 Matplotlib 库中的 scatter() 函数，并为每个数据点指定一种颜色。

以下是一个使用 scatter() 函数实现四种不同数值的数据不同颜色散点图的示例代码：

import numpy as np
import matplotlib.pyplot as plt
 
# 创建数据集
x = np.random.rand(50)
y = np.random.rand(50)
z = np.random.randint(1, 5, 50)
 
# 设置颜色映射
color_map = plt.cm.get_cmap('Set1', 4)
 
# 设置绘图参数
plt.scatter(x, y, s=100, c=z, cmap=color_map)
plt.colorbar(ticks=[1, 2, 3, 4])
plt.title('Scatter Plot')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
 
# 显示图形
plt.show()

在这个例子中，我们创建了一个包含 50 个数据点的数据集，其中 x 和 y 分别表示数据点的 x 和 y 坐标，z 表示每个数据点的值，这里我们使用 np.random.randint() 函数生成 1-4 的整数。

我们使用 plt.cm.get_cmap() 函数获取 'Set1' 颜色映射，并将其用于设置四种不同数值的数据点的颜色。我们通过设置 ticks 参数指定 colorbar 显示的颜色对应的数值。

最后，我们使用 scatter() 函数绘制散点图，并使用 colorbar() 函数显示颜色映射。