mercredi 15 juin 2016

Assigning 2d array in vector of indices


Given 2d array k = np.zeros((M, N)) and list of indices in the range 0, 1 .., M-1 of size N called places = np.random.random_integers(0, M-1, N) how do I assign 1 in each column of k in the places[i] index where i is running index. I would like to achieve that in python compact style and without any loops

Examples:

N = 5, M =3
places= 0, 0, 1, 1, 2

Then:

k = [1, 1, 0, 0, 0
     0, 0, 1, 1, 0
     0, 0, 0, 0, 1]

Aucun commentaire:

Enregistrer un commentaire