python DataFrame onehot编码，一列变多列

最新推荐文章于 2024-02-26 20:09:39 发布

genghaihua

最新推荐文章于 2024-02-26 20:09:39 发布

阅读量3.7k

点赞数

分类专栏： python

本文链接：https://blog.csdn.net/genghaihua/article/details/89963596

版权

python 专栏收录该内容

20 篇文章

订阅专栏

博客围绕Python中DataFrame的OneHot编码展开，介绍了将一列数据转换为多列的操作。OneHot编码是数据处理的重要手段，在数据分析和机器学习等领域有广泛应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import pandas as pd
carria_code_dict={"CZ":0,"SC":1,"Y8":2,"3U":3,"MU":4,"NS":5}
carria_len=carria_code_dict.__len__()
def encode_data(str):
    datas=str.split(',')
    encode_result=['0']*carria_len
    for data in datas:
        temp=data.split(":")
        if(temp.__len__()!=2 or  not carria_code_dict.has_key(temp[0])):
            pass
        else:
            encode_result[carria_code_dict.get(temp[0])]=temp[1]
    return ','.join(encode_result)
df2=pd.DataFrame([["CZ:1,SC:4,Y8:1"],["3U:2,CZ:1,MU:6,NS:3"]],
                 columns=['carrier'])
df2[carria_code_dict.keys()] =df2['carrier'].map(lambda x:encode_data(x)).str.split(',',expand=True)
print df2