[python] How to do multiple arguments to map function where one remains the same in python?