Scale(or normalize) an array like this in numpy?

Question

In numpy, the original array has the shape(2,2,2) like this

[[[0.2,0.3],[0.1,0.5]],[[0.1,0.3],[0.1,0.4]]]

I'd like to scale the array so that the max value of the a dimension is 1 like this:

As max([0.2,0.1,0.1,0.1]) is 0.2, and 1/0.2 is 5, so for the first element of the int tuple, multiple it by 5.

As max([0.3,0.5,0.3,0.4]) is 0.5, and 1/0.5 is 2, so for the second element of the int tuple, multiple it by 2

So the final array is like this:

[[[1,0.6],[0.5,1]],[[0.5,0.6],[0.5,0.8]]]

I know how to multiple an array with an integer in numpy, but I'm not sure how to multiple the array with different factor. Does anyone have ideas about this?

Lee · Accepted Answer · 2013-06-25 08:49:17Z

4

If your array = a:

>>> import numpy as np
>>> a = np.array([[[0.2,0.3],[0.1,0.5]],[[0.1,0.3],[0.1,0.4]]])

You can do this:

>>> a/np.amax(a.reshape(4,2),axis=0)
array([[[ 1. ,  0.6],
        [ 0.5,  1. ]],

       [[ 0.5,  0.6],
        [ 0.5,  0.8]]])

answered Jun 25, 2013 at 8:49

Lee

31.4k31 gold badges124 silver badges187 bronze badges

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

Collectives™ on Stack Overflow

Scale(or normalize) an array like this in numpy?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related