Sentence examples for a punishment from inspiring English sources

"a punishment" is correct and usable in written English.
You can use it to refer to an action taken by someone to discipline someone else, usually in response to an infraction or wrongdoing. For example, "The principal gave me a punishment of after-school detention for being late to class."