Build a model using only pandas/numpy | Nextdoor Interview Question