python提取csv文件的特征集和label集代码
时间: 2024-05-13 16:18:33 浏览: 200
python实现对csv文件的列的内容读取
假设你的CSV文件中第一列是标签列,剩余列是特征列,可以使用以下代码来提取特征集和标签集:
```python
import pandas as pd
# 读取CSV文件
data = pd.read_csv('your_file.csv')
# 提取标签列
label_col = data.iloc[:, 0]
label = label_col.values
# 提取特征列
features_col = data.iloc[:, 1:]
features = features_col.values
```
其中,`iloc` 函数用于按位置选择行和列,第一个参数指定行,第二个参数指定列。`values` 属性用于将数据转换为 NumPy 数组。
阅读全文