Axion Ray interview question

Coding in python for visualisation. Code in python for some data manipulation