Trains that emit pure water could be in the UK by the “early 2020s”, according to the government.