how to select a rndom number of inputs from an array to put in another array python code example

Example: numpy random choice

>>> np.random.choice(5, 3, p=[0.1, 0, 0.3, 0.6, 0])
array([3, 3, 0])