Efficient numerical modelling of the emittance evolution of beams with finite energy spread in plasma wakefield accelerators