Zambian President Edgar Lungu has been discharged from hospital in South Africa after undergoing surgery to correct a narrowing of the oesophagus.