操作pandas某一列实现one-hot

最新推荐文章于 2023-03-01 18:01:57 发布

ChaoFeiLi

最新推荐文章于 2023-03-01 18:01:57 发布

阅读量1k

点赞数

分类专栏： Python

本文链接：https://blog.csdn.net/ChaoFeiLi/article/details/115345237

版权

Python 专栏收录该内容

41 篇文章 0 订阅

订阅专栏

本文是转载，但是忘记转载的链接是那个了。非常抱歉。

def ontHot():
    import pandas as pd
    df = pd.DataFrame([
        ['green', 'M', 10.1, 'class1'],
        ['red', 'L', 13.5, 'class2'],
        ['blue', 'XL', 15.3, 'class1']])

    df.columns = ['color', 'size', 'prize', 'class label']
    size_mapping = {
        'XL': 3,
        'L': 2,
        'M': 1}
    df['size'] = df['size'].map(size_mapping)

    class_mapping = {label: idx for idx, label in enumerate(set(df['class label']))}
    df['class label'] = df['class label'].map(class_mapping)
    return df