How to use lambda with Python