Assign a color to a column in a data frame based on color map from `matplotlib`

Question

I want to assign a color to each value of the column df[\'B\'] based on a color map from matplotlib.

df = pd.DataFrame({"A": [