Not a book, but I recommend as a starting point the Farnam Street blog: https://www.farnamstreetblog.com/mental-models/
* Mental Models: The Best Way to Make Intelligent Decisions (113 Models Explained) - https://www.farnamstreetblog.com/mental-models/