Not a book, but I have found Trey Hunner's https://www.pythonmorsels.com/ exercises very useful on this front.

The solutions presented typically include both a "basic" approach, a "as pythonic as possible" approach, and a brief discussion of the trade-offs between elegance and readability, etc.