Reddit Interview Question

Simple ML coding question with data