Numpy Random Choice Index