Replacing for-loop in python when index is part of a array-manipulating function

Question

I have a multidimensional numpy array which I need to modify such that its elements are modified as a function of the index of one of the dimensions only. I can of course do that with a for loop, as in the following simplified example

import numpy as np
a = np.ones( (2,10) )
for ii in range(a.shape[1]):
    a[:,ii] *= ii

If the array becomes very large, this might slow down the execution and I was wondering if there are some clever ways to avoid using a for loop?

Mateen Ulhaq · Accepted Answer · 2021-11-18 22:57:40Z

1

Construct another array to hold the scaling factors, then broadcast and multiply:

scale = np.arange(a.shape[1])
a *= scale

answered Nov 18, 2021 at 22:57

Mateen Ulhaq

27.8k21 gold badges121 silver badges155 bronze badges

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

Collectives™ on Stack Overflow

Replacing for-loop in python when index is part of a array-manipulating function

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