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)
原文地址:https://www.cnblogs.com/songyuejie/p/14789508.html