python读取csv文件前两列作为特征,第三列作为标签,并画图。在图上空白处显示颜色对应类别,用横线表示
时间: 2024-02-21 18:56:40 浏览: 16
好的,你想用横线表示什么呢?如果你想用横线将数据分隔开来,可以使用以下代码:
```python
import pandas as pd
import matplotlib.pyplot as plt
# 读取csv文件
df = pd.read_csv('your_file.csv')
# 获取前两列作为特征,第三列作为标签
X = df.iloc[:, :2].values
y = df.iloc[:, 2].values
# 将数据分成两部分
X1 = X[y == 0]
X2 = X[y == 1]
# 绘制散点图
plt.scatter(X1[:, 0], X1[:, 1], c='blue', label='Class 0')
plt.scatter(X2[:, 0], X2[:, 1], c='red', label='Class 1')
# 添加横线
plt.axhline(y=0.5, color='black', linestyle='--')
# 显示图例
plt.legend()
# 显示图形
plt.show()
```
这段代码会读取名为“your_file.csv”的csv文件,获取前两列作为特征,第三列作为标签,并绘制散点图。将标签为0和1的数据分成两部分,分别用蓝色和红色的散点图表示。添加一条横线,将数据分隔开来。在图例中显示不同类别的颜色。