Assign values to numpy.array

Question

Let say we create this np.array:

A = np.arange(12).reshape(3, 4)

So, A is:

array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

I would like to assign some values to A like that:

A[[0,2]][:,[1,3]] = 9999

In order to obtain:

array([[ 0,  9999,  2,  9999],
       [ 4,  5,  6,  7],
       [ 8,  9999, 10, 9999]])

But this doesn't work. What is the proper way to do it?

Thanks

Divakar · Accepted Answer · 2017-04-05 11:22:10Z

1

You can use np.ix_ to get those open meshes, which when used for indexing into the array would be broadcasted and thus could be used for assigning values into it, like so -

A[np.ix_([0,2],[1,3])] = 9999

answered Apr 5, 2017 at 11:22

Divakar

222k19 gold badges273 silver badges374 bronze badges

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

Collectives™ on Stack Overflow

Assign values to numpy.array

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related