{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"pygments_lexer":"ipython3","nbconvert_exporter":"python","version":"3.6.4","file_extension":".py","codemirror_mode":{"name":"ipython","version":3},"name":"python","mimetype":"text/x-python"}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"import numpy as np\nimport pandas as pd\n\nTRAIN_PATH = \"../input/house-prices-advanced-regression-techniques/train.csv\"\n\nID = \"Id\"\nTARGET = \"SalePrice\"\n\ntrain = pd.read_csv(TRAIN_PATH)","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","execution":{"iopub.status.busy":"2022-07-26T22:26:14.747319Z","iopub.execute_input":"2022-07-26T22:26:14.748467Z","iopub.status.idle":"2022-07-26T22:26:14.776816Z","shell.execute_reply.started":"2022-07-26T22:26:14.748414Z","shell.execute_reply":"2022-07-26T22:26:14.775911Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"def getFeatureImportance(df,TARGET,num = -1):\n    # correlation \n    corrTrain = df.corr()\n    \n    # abs \n    corrTrain = np.abs(corrTrain)\n    \n    # to dataframe \n    corrTarget = corrTrain[[TARGET]]\n    \n    # sort \n    corrTarget = corrTarget.sort_values(TARGET,ascending=False)\n    corrTargetT = corrTarget.T\n    \n    # delete target\n    corrTargetTImprortance = [col for col in corrTargetT.columns if col != TARGET and col != ID]\n    \n    # select feature importance top num\n    if num != -1:\n         selCorrTargetTImprortance = corrTargetTImprortance[:num]\n    else:\n        selCorrTargetTImprortance = corrTargetTImprortance\n        \n    result = corrTargetT[selCorrTargetTImprortance]\n    \n    return result.T","metadata":{"execution":{"iopub.status.busy":"2022-07-26T22:26:14.893926Z","iopub.execute_input":"2022-07-26T22:26:14.894565Z","iopub.status.idle":"2022-07-26T22:26:14.903863Z","shell.execute_reply.started":"2022-07-26T22:26:14.894520Z","shell.execute_reply":"2022-07-26T22:26:14.902566Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"result = getFeatureImportance(train,TARGET)\nresult","metadata":{"execution":{"iopub.status.busy":"2022-07-26T22:26:14.905955Z","iopub.execute_input":"2022-07-26T22:26:14.906966Z","iopub.status.idle":"2022-07-26T22:26:14.933858Z","shell.execute_reply.started":"2022-07-26T22:26:14.906912Z","shell.execute_reply":"2022-07-26T22:26:14.932778Z"},"trusted":true},"execution_count":null,"outputs":[]}]}