Convert Categories Feature to numbers

import pandas as pd


if __name__ == '__main__':
    mapping = {
        "Class0": 0,
        "Class1": 1,
        "Class2": 2,
        "Class3": 3,
        "Class4": 4,
        "Class5": 5
    }
    df = pd.read_csv("../input/train.csv")
    # df.loc[:, "catagory"] = df.catagory.map(mapping)
    df.loc[:, "catagory"] = df["catagory"].map(mapping)

标签：__,loc,Convert,df,Feature,mapping,catagory,csv,Categories
来源： https://www.cnblogs.com/songyuejie/p/14789508.html