Pandas 数据提取与分类散点图绘制：基于前两维特征
假设数据集为 DataFrame 类型，前两维特征为'feature1'和'feature2'，分类标签为'label'：
- 提取前两维特征
X = data[['feature1', 'feature2']]
- 提取分类标签
y = data['label']
- 绘制散点图
import matplotlib.pyplot as plt
# 根据分类标签划分数据集
X1 = X[y==0]
X2 = X[y==1]
# 绘制散点图
plt.scatter(X1.iloc[:,0], X1.iloc[:,1], label='Class 0')
plt.scatter(X2.iloc[:,0], X2.iloc[:,1], label='Class 1')
plt.legend()
plt.show()
原文地址: https://www.cveoy.top/t/topic/mNCn 著作权归作者所有。请勿转载和采集!