I was about to suggest Astral City and Miss Hepburn beat me to it!
Spiritist religion (its brother in the northern hemisphere is called Spiritualism) is very much based on this idea that you go where your conscious vibration is best suited. The Astral City is largely based on that.
More spiritist literature includes "The Spirits Book" and "Heaven and Hell". All freely available online in pdf.s