Customize container prompt

I have installed docker on local Ubuntu machine. I have created 3 different containers of one single image.
I am trying to log into container using below command
container 1 --> docker exec -it ubuntu01 bash
container 2 --> docker exec -it ubuntu02 bash
container 3 --> docker exec -it ubuntu03 bash

For all above 3 I am getting different command . sample output given below
root@42d3a67c0526
root@55d3a67c0545
root@42d3a70c0527

How to customize above, so that i would be able to know to which container i am working with?

You can assign “–hostname” flag, when you are creating docker container with docker run command.