文字分類轉數字代號 – 柯博文老師


#!/usr/bin/env python
# -*- coding=utf-8 -*-
# pip install seaborn
__author__ = "柯博文老師 Powen Ko, www.powenko.com"
 
import xlrd
import xlwt

import pandas as pd

 

#read data
df = pd.read_csv('adult.csv')

print(df.head())
print(df.columns)
print(df.index)
print(df.columns)
 
columnsName=['B','D','F','G','H','I','J','N','O']

for x in columnsName:
    df[x+'_Code'] = df[x].astype("category").cat.codes

#df['Spectral_Class_Code'] = df['Spectral_Class'].astype("category").cat.codes

print(df.head())

df.to_excel("adult.xlsx")
Python
