Np.choice Python