data preprocessing python script