python绘制分类的散点图
时间: 2023-07-11 15:52:39 浏览: 50
要绘制分类的散点图,可以使用Python的Matplotlib库。假设有两个类别(A和B)的数据,可以按照以下步骤绘制分类的散点图:
1. 导入Matplotlib库和NumPy库(如果需要):
```
import matplotlib.pyplot as plt
import numpy as np
```
2. 创建两个类别的数据:
```
x1 = np.random.normal(0, 1, 100) # 类别A的x坐标
y1 = np.random.normal(0, 1, 100) # 类别A的y坐标
x2 = np.random.normal(2, 1, 100) # 类别B的x坐标
y2 = np.random.normal(2, 1, 100) # 类别B的y坐标
```
3. 绘制散点图:
```
plt.scatter(x1, y1, c='r', marker='o', label='A') # 类别A的散点图
plt.scatter(x2, y2, c='b', marker='s', label='B') # 类别B的散点图
plt.legend(loc='upper right') # 添加图例
plt.show() # 显示图形
```
完整的代码如下:
```
import matplotlib.pyplot as plt
import numpy as np
# 创建两个类别的数据
x1 = np.random.normal(0, 1, 100) # 类别A的x坐标
y1 = np.random.normal(0, 1, 100) # 类别A的y坐标
x2 = np.random.normal(2, 1, 100) # 类别B的x坐标
y2 = np.random.normal(2, 1, 100) # 类别B的y坐标
# 绘制散点图
plt.scatter(x1, y1, c='r', marker='o', label='A') # 类别A的散点图
plt.scatter(x2, y2, c='b', marker='s', label='B') # 类别B的散点图
plt.legend(loc='upper right') # 添加图例
plt.show() # 显示图形
```