`a = np.ones(10); a[np.arange(10)] = [1, 2, 3]`