Axion Ray Interview Question

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