How can I replace map() function with list comprehension or for loop