心脏病题目（2）数据预处理

最新推荐文章于 2023-12-20 16:55:04 发布

CristinaM

最新推荐文章于 2023-12-20 16:55:04 发布

阅读量1.5k

点赞数 1

本文链接：https://blog.csdn.net/weixin_43615654/article/details/89603063

版权

一、幅度调整，归一化

Age，Trestbps，chol，thalach，oldpeak（连续参数）属于数据比较集中的类型，可用MAXMIN进行标准化处理。

from sklearn import preprocessing
col = data[['age','trestbps','chol','thalach','oldpeak']]
min_max_scaler = preprocessing.MinMaxScaler()
col_min_max = min_max_scaler.fit_transform(col)
col_min_max = pd.DataFrame(col_min_max)
col_min_max.columns = ['age','trestbps','chol','thalach','oldpeak']

cp，slope，restecg，thal进行onehot编码

OneHot_cp = pd.get_dummies(data.cp,prefix='cp').astype('float')
OneHot_restecg = pd.get_dummies(data.restecg,prefix='restecg').astype('float')
OneHot_slope = pd.get_dummies(data.slope,prefix='slope').astype('float')
OneHot_ca = pd.get_d