python读取csv文件前两列作为特征,第三列作为标签,并画图。在图上空白处显示颜色对应类别
时间: 2024-02-21 10:56:33 浏览: 51
python读取csv文件的几种方式(含实例说明)
好的,我可以帮你解决这个问题。首先,你需要安装以下包:pandas, matplotlib。
然后,你可以使用以下代码来读取csv文件并绘制散点图:
```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
# 绘制散点图
plt.scatter(X[:, 0], X[:, 1], c=y)
# 显示颜色对应类别
plt.colorbar()
# 显示图形
plt.show()
```
这段代码会读取名为“your_file.csv”的csv文件,获取前两列作为特征,第三列作为标签,并绘制散点图。在图的右侧会显示颜色对应的类别。
阅读全文