import numpy as np

# %% Params of choice():

sample_space = np.array(['Red', 'Blue', 'Green'])
sample_size = 2
replace = False
prob = [1/3, 1/3, 1/3]

# %% Generating random sample:

sample = np.random.choice(sample_space,
                          sample_size,
                          replace,
                          prob)

print('sample =', sample)