Some of the best books for problem solving in Python are-

For Beginner level the books you can follow-

• Learn Python the hard way
• Automate the boring stuff

For Intermediate level the book which will alone sufficient is Problem solving with Algorithms and Data structure using Python by Brad Miller and David Ranum, Luther College.

I hope this helps. Wish you a great day!