Fill numpy array by indexing with 2d array

Question

I have a 2d numpy array of zeros subbins, and a 2d numpy array of indices into it combos. For example

p = 4
combos = np.asarray(list(itertools.combinations(range(p),3)))
subbins = np.zeros(shape=(len(combos),p))

The arrays look like this

combos = [
     [0, 1, 2],
     [0, 1, 3],
     [0, 2, 3],
     [1, 2, 3],
]
subbins = [
     [0, 0, 0, 0],
     [0, 0, 0, 0],
     [0, 0, 0, 0],
     [0, 0, 0, 0],
]

How can I use combos to index into subbins and assign values without iterating - as pythonic as possible? I.e. the output I want is this:

output = [
     [1, 1, 1, 0],
     [1, 1, 0, 1],
     [1, 0, 1, 1],
     [0, 1, 1, 1],
]

yatu · Accepted Answer · 2020-02-19 11:14:28Z

2

We can use np.put_along_axis:

np.put_along_axis(subbins, combos, 1, axis=1)

print(subbins)
array([[1., 1., 1., 0.],
       [1., 1., 0., 1.],
       [1., 0., 1., 1.],
       [0., 1., 1., 1.]])

answered Feb 19, 2020 at 11:14

yatu

88.6k12 gold badges93 silver badges148 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Fill numpy array by indexing with 2d array

1 Answer 1

Comments

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Related