Skip to content

Instantly share code, notes, and snippets.

@QuantumDamage

QuantumDamage/dummies.py

Created July 12, 2017 04:46
Show Gist options
  • Save QuantumDamage/4619f5e20f295d6b2aa3b71f6ef5e7b5 to your computer and use it in GitHub Desktop.
Save QuantumDamage/4619f5e20f295d6b2aa3b71f6ef5e7b5 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
dummies.py
import pandas as pd
import numpy as np
dataTrain = pd.DataFrame({'county': ['Cochice', 'Pima', 'Santa Cruz', 'Maricopa', 'Yuma']})
dummiesTrain = pd.get_dummies(dataTrain)
dataLive = pd.DataFrame({'county': ['Santa Cruz','Poznań', 'Pima']})
dummiesLive = pd.get_dummies(dataLive)
dummiesLive
dummiesTrain
dummiesLive.reindex(columns = dummiesTrain.columns, fill_value=0)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment