在Python或Sklearn中使用整数值对具有字符串值的列变量进行编码

如何用整数值编码数据表中字符串类型的列值.例如,我有两个特征变量：颜色(可能的字符串值R,G和B)和技能(可能的字符串值C,Java,sql和Python).给定的数据表有两列-

Color' -> R G B B G R B G G R G  ;
Skills' -> Java , C++, sql, Java, Python, Python, sql, C++, Java, sql, Java.

我想知道哪种sklearn函数/方法将在以上两列上进行转换,如R = 0,G = 1和B = 2以及C = 0,Java = 1,sql = 2和Python = 3：

Color: 0, 1, 2, 2, 1, 0, 2, 1, 1, 0, 1
Skills:  1, 0, 2, 1, 3, 3, 2, 0, 1, 2, 1

请让我知道该怎么做？

解决方法:

使用Sckit学习LabelEncoder()方法

import pandas as pd
from sklearn.preprocessing import LabelEncoder

df = pd.DataFrame({
'colors':  ["R" ,"G", "B" ,"B" ,"G" ,"R" ,"B" ,"G" ,"G" ,"R" ,"G" ],
'skills':  ["Java" , "C++", "sql", "Java", "Python", "Python", "sql","C++", "Java", "sql", "Java"]
})

def encode_df(dataframe):
    le = LabelEncoder()
    for column in dataframe.columns:
        dataframe[column] = le.fit_transform(dataframe[column])
    return dataframe

#encode the dataframe
encode_df(df)