A simple code for query the CSV data with python