Skip to content

Instantly share code, notes, and snippets.

@aniruddha27

aniruddha27/House_Price_Features.py

Created March 17, 2020 06:15
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save aniruddha27/876953c06de00180160460df3eb8b0e6 to your computer and use it in GitHub Desktop.
Save aniruddha27/876953c06de00180160460df3eb8b0e6 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
House_Price_Features.py
# Total surface area of house
train['TotalSF'] = train.apply(lambda x: x['1stFlrSF'] + x['2ndFlrSF'] + x['TotalBsmtSF'], axis=1)
# Total number of bathrooms in the house
train['TotalBath'] = train.apply(lambda x: x['FullBath'] + 0.5*x['HalfBath'] + x['BsmtFullBath'] + 0.5*x['BsmtHalfBath'], axis=1)
# Total Porch area in the house
train['TotalPorch'] = train.apply(lambda x: x['OpenPorchSF'] + x['EnclosedPorch'] + x['3SsnPorch'] + x['ScreenPorch'], axis=1)
# New house or an old house
train['NewHouse'] = train.apply(lambda x: 1 if x['SaleCondition']=='Partial' else 0, axis=1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment