My kids understood different contexts around four or five. Every kid is different of course, but the explanation for why we don’t swear in public was just “it makes some people really uncomfortable, so we don’t use those words in public to be nice to them”.

They still slip up once in a while, but hell, so do I!