pandas.get_dummies 的用法

最新推荐文章于 2024-04-21 20:59:44 发布

风雪云侠

最新推荐文章于 2024-04-21 20:59:44 发布

阅读量371

点赞数

分类专栏： python问题文章标签： numpy python

本文链接：https://blog.csdn.net/weixin_40426830/article/details/108693702

版权

python问题专栏收录该内容

6 篇文章 1 订阅

订阅专栏

pandas.get_dummies 的用法

get_dummies 是利用pandas实现one hot encode的方式。

import numpy as np 
import pandas as pd 
import matplotlib.pyplot as plt
import torch 
from torch import nn 
from torch.utils.data import Dataset,DataLoader,TensorDataset

dftrain_raw = pd.read_csv('/content/eat_pytorch_in_20_days/data/titanic/train.csv')
dftest_raw = pd.read_csv('/content/eat_pytorch_in_20_days/data/titanic/test.csv')
dftrain_raw.head(10)

Pclass:乘客所持票类，有三种值(1,2,3) 【转换成onehot编码】
在这里插入图片描述

    dfdata = dftrain_raw
    dfresult= pd.DataFrame()
    #Pclass
    dfPclass = pd.get_dummies(dfdata['Pclass'])

get_dummies 后：在这里插入图片描述

    dfPclass.columns = ['Pclass_' +str(x) for x in dfPclass.columns ]
    dfresult = pd.concat([dfresult,dfPclass],axis = 1)
    dfresult

在这里插入图片描述
Embarked:乘客登船港口:S、C、Q(有缺失)【转换成onehot编码，四维度 S,C,Q,nan】

pd.get_dummies(dfdata['Embarked'],dummy_na=True)

在这里插入图片描述

dfEmbarked = pd.get_dummies(dfdata['Embarked'],dummy_na=True)
dfEmbarked.columns = ['Embarked_' + str(x) for x in dfEmbarked.columns]
dfresult = pd.concat([dfresult,dfEmbarked],axis = 1)

在这里插入图片描述