Found in 1 comment on Hacker News
https://www.amazon.com/Mental-Models-Cognitive-Inference-Con... Quite old, but probably the best book about mental models

Fresh book recommendations delivered straight to your inbox every Thursday.