Python Bubble Maps with Different Coloring for U.S. Regions (Midwest, Southeast,...)

[sgunpinar]

I am currently using 'albers usa' in geo projection and plotly package to create bubble maps in Python. The code is shown in the below link.
https://plot.ly/python/bubble-maps/

Is there any way to customize the land color in the map based on five U.S. regions? The code accommodates only one land color. If any help provided to revise the existing code or to handle this issue, I will appreciate it. Thank you.