Formatting every element in a list based on some function in python