The Obama administration is buying 32 tons of heavy water, a key component in atomic-weapons development, from Iran in an effort to safeguard its landmark nuclear agreement with the country, according to senior American officials.

The Department of Energy’s impending purchase was driven by U.S. concerns that Iran doesn’t have the capacity yet to quickly reduce its stockpile of the material as required under the July…