Transform columns in a csv using python