不具备顺序的离散变量常用独热编码,具备顺序的常用标签编码
import pandas as pd
data = pd.read_csv('data.csv')
data["Home Ownership"].value_counts()#找一列试试,但这一列也可以独热编码,可以都运行一下试试哪个更好
# 按贷款严重程度定义映射字典
mapping = {
"Rent": 0,
"Own Home": 1,
"Have Mortgage ": 2,
"Home Mortgage": 3
}
data["Home Ownership"].head()
data["Home Ownership"] = data["Home Ownership"].map(mapping)
data["Home Ownership"].head()
data["Term"].value_counts()
# 定义映射字典,两类型的也整成整数,不过三类型以上才是独热编码或标签编码之类的
mapping = {
"Short Term": 1,
"Long Term": 0
}
# 进行映射
data["Term"] = data["Term"].map(mapping)
data["Term"].head()