Python code taking information from a .csv file and return a matrix then printing results by user choice.