Создать новый массив с элементами массива / суммами, установленными двумя другими массивами в Python

Question

У меня есть два массива в Python (numpy массивы):

a=array([5,7,3,5])
b=array([1,2,3,4])

и я хочу создать третий массив с каждым элементом из b, появляющимся a раз в новом массиве, как:

c=array([1,1,1,1,1,2,2,2,2,2,2,2,3,3,3,4,4,4,4,4])

Существует ли быстрый, numPythonic способ сделать это с минимумом циклов? Мне нужно использовать эту операцию тысячи раз в цикле над довольно большим массивом, поэтому я хотел бы, чтобы она была максимально быстрой.

Cheers, Mike

Answer 1

Я верю повтор это то, что вы хотите:

c = repeat(b, a)