Toisen maailmansodan lopussa vuonna 1945 Pu Yi joutui puna-armeijan vangiksi. Kun Mao Zedongin johtamat kommunistit ottivat vallan Kiinassa vuonna 1949, Pu Yi kirjoitti kirjeitä Neuvostoliiton diktaattori Josif Stalinille, joissa hän vakuutti sisäistäneensä kommunismin ja pyysi asuinpaikkaa Neuvostoliitosta. Stalin kuitenkin luovutti Pu Yin kiinalaisille kommunisteille vuonna 1950. Sitä ennen Pu Yi oli ollut todistajana Tokion kansainvälisessä sotarikostuomioistuimessa. Hän vietti kymmenen vuotta uudelleenkoulutusleirillä Fushunissa, ja sen jälkeen hänen ilmoitettiin uudistuneen ja ryhtyneen kommunismin kannattajaksi. Vapautuksensa jälkeen hän työskenteli Pekingin kasvitieteellisessä puutarhassa ja vuodesta 1964 Kiinan kansan neuvoa-antavan poliittisen konferenssin jäsenenä. Pu Yin viides vaimo oli Li Shuxian, jonka kanssa hän avioitui vuonna 1962. Li kuoli vuonna 1997.