NVIDIA Interview Question

The interviewer asked me to use Google Docs to complete a flood-fill algorithm based on a numpy input image and starting location. Then, I had to adapt it to make a list of the pixels surrounding a starting location that shared the same color.