Numpy scale specific values of numpy array

Question

I want to scale specific values (e.g. lager than 5) of an numpy array numbers by a multiplier (e.g. 2). I know I can achieve this by a loop, but I wanted to avoid loops. I think I can achieve this somehow with a numpy mask but I am not sure how to implement it. In order to demonstrate what I aim to achieve I used the imaginary function scale_array.

Here is a minimal working example

import numpy as np

numbers = np.array([-3, 5, 2, -1, -15, 10])
mask = np.abs(numbers) > 5

numbers_scaled = scale_array(array=numbers, mask=mask, scale_factor=2)
print(numbers_scaled)  # np.array([-3, 5, 2, -1, -30, 20])

Space Impact · Accepted Answer · 2018-09-18 09:33:30Z

1

Directly assign masked values multiplied by 2 to original array as:

numbers = np.array([-3, 5, 2, -1, -15, 10])
mask = np.abs(numbers) > 5

numbers[mask] = numbers[mask]*2
numbers

array([ -3,   5,   2,  -1, -30,  20])

answered Sep 18, 2018 at 9:33

Space Impact

13.3k26 silver badges51 bronze badges

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

1 Comment

New2Coding Over a year ago

Thank you for your help! That was exactly what I was looking for.

PAb · Accepted Answer · 2018-09-18 09:33:44Z

1

As you write, mask = np.abs(numbers) > 5 gives you the locations you want to scale.

Simply doing numbers[mask] *= 2 should do the trick :)

answered Sep 18, 2018 at 9:33

PAb

5612 silver badges10 bronze badges

Collectives™ on Stack Overflow

Numpy scale specific values of numpy array

2 Answers 2

1 Comment

Comments

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Related