Python画散点图,不同点画不同颜色散点图是数据可视化中常用的一种图形,它可以展示数据的分布情况以及数据点之间的关系。在Python中,通过matplotlib库的scatter函数可以轻松绘制散点图。本文将从多个角度分析如何使用Python画散点图,并实现不同点画不同颜色的效果。
1. 数据准备
在绘制散点图之前,需要先准备数据。假设有一组数对(x1, y1), (x2, y2), ..., (xn, yn),我们要将这些数据点展示在散点图上。可以使用numpy库生成一组随机数据,代码如下:
```python
import numpy as np
# 生成随机数据
n = 100
x = np.random.rand(n)
y = np.random.rand(n)
```
其中,np.random.rand(n)可以生成n个0到1之间的随机数。
2. 绘制散点图
使用matplotlib库的scatter函数绘制散点图。代码如下:
```python
import matplotlib.pyplot as plt
# 绘制散点图
plt.scatter(x, y)
plt.show()
```
运行代码,会弹出一个新窗口显示绘制出来的散点图。如下图所示:
![散点图1](https://img-blog.csdn.net/20180401190712972)
可以看到,散点图上展示了100个数据点的分布情况。
3. 不同点画不同颜色
有时候,我们需要将散点图上不同的数据点用不同的颜色来表示。例如,在某个实验中,我们需要将实验结果分成两类,分别用红色和蓝色的散点来表示。可以通过给scatter函数传递一个表示颜色的参数c来实现这个效果。代码如下:
```python
# 生成随机标签,0或1
labels = np.random.randint(0, 2, n)
# 根据标签绘制不同颜色的散点图
plt.scatter(x, y, c=labels)
plt.show()
```
其中,np.random.randint(0, 2, n)可以生成n个0或1的随机数,用来表示数据点的标签。
运行代码,会弹出一个新窗口显示绘制出来的散点图。如下图所示:
![散点图2](https://img-blog.csdn.net/20180401190838458)
可以看到,散点图上展示了100个数据点,并且将标签为0的点用蓝色表示,标签为1的点用红色表示。
4. 自定义颜色
除了使用默认的颜色之外,还可以自定义颜色。可以通过给scatter函数传递一个数组c来实现这个效果。数组c中的每个元素代表一个数据点的颜色,可以是RGB颜色值,也可以是颜色名称。例如,下面的代码将散点图上的数据点分成了三类,分别用蓝色、绿色和红色表示。
```python
# 生成随机标签,0、1或2
labels = np.random.randint(0, 3, n)
# 自定义颜色
colors = ['b', 'g', 'r']
# 根据标签绘制不同颜色的散点图
plt.scatter(x, y, c=[colors[label] for label in labels])
plt.show()
```
运行代码,会弹出一个新窗口显示绘制出来的散点图。如下图所示:
![散点图3](https://img-blog.csdn.net/20180401191015486)
可以看到,散点图上展示了100个数据点,并且将标签为0的点用蓝色表示,标签为1的点用绿色表示,标签为2的点用红色表示。
5. 结语
本文介绍了如何使用Python绘制散点图,并实现不同点画不同颜色的效果。通过使用numpy库生成随机数据,并使用matplotlib库的scatter函数绘制散点图,可以轻松地展示数据的分布情况和数据点之间的关系。同时,通过给scatter函数传递颜色参数,可以实现不同点画不同颜色的效果。这些技巧在数据可视化中非常实用,希望对读者有所帮助。