The colors observed during the flame test result from the excitement of the electrons caused by the increased temperature. The electrons "jump" from their ground state to a higher energy level. As they return to their ground state, they emit visible light.
