Well, if you assume two things, yes.

First, that prison serves a purpose beyond punishment. Second, that there’s no better option to achieve that purpose.

Without those assumptions, then you get different answers because if punishment is the only purpose of it, there’s way less expensive and societally detrimental ways to punish someone. And, if there’s no better options to achieve other goals, then you have to wonder if the goals are worthwhile.