Retrieve JSON from CSV using python in desired format