Create array X with certain number of copies of each element in array Y in python [duplicate]

Question

I have an array counts = [2, 3, 4] with each element denoted by c. I want to convert it into an array weights that has m copies of each element c, where m is an element in M = [3, 2, 1].

So in the end, I'd like an array weights = [2, 2, 2, 3, 3, 4]

Is there an efficient way to do this in numpy?

Mad Physicist · Accepted Answer · 2020-05-13 00:59:55Z

4

numpy.repeat does exactly what you want:

>>> np.repeat(counts, M)
array([2, 2, 2, 3, 3, 4])

answered May 13, 2020 at 0:59

Mad Physicist

116k29 gold badges202 silver badges292 bronze badges

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

Collectives™ on Stack Overflow

Create array X with certain number of copies of each element in array Y in python [duplicate]

1 Answer 1

Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Linked

Related