tmp2 = np.array(['white', 'black', 'blue', 'blue'])
color1_occupations = [1, 0, 1, 2]
color2_occupations = [0, 2, 0, 1]
using numpy repeat simultaneously on arrays with distinct multiplicities but same dimension