导入数据对所有分类数据的列进行编码python
时间: 2023-07-14 20:14:11 浏览: 79
可以使用sklearn中的LabelEncoder进行分类数据的编码。以下是一个示例代码:
```python
from sklearn.preprocessing import LabelEncoder
# 读取数据
data = pd.read_csv('data.csv')
# 获取所有分类数据列
cat_cols = [col for col in data.columns if data[col].dtype == 'object']
# 对每一列进行编码
for col in cat_cols:
le = LabelEncoder()
data[col] = le.fit_transform(data[col])
```
在上述代码中,我们首先使用pandas库读取数据。然后,我们使用列表推导式获取所有数据类型为“object”的列,并将其存储在cat_cols变量中。
接下来,我们使用for循环对每个分类数据列进行编码。我们使用LabelEncoder来进行编码。最后,我们将编码后的列替换回原始数据集中的列。
阅读全文