创建数组X，并在python中为数组Y中的每个元素提供一定数量的副本

Posted 2021-04-30

我有一个数组counts = [2, 3, 4]，每个元素都用c表示。我想将其转换为具有每个元素weights的m个副本的数组c，其中m是M = [3, 2, 1]中的元素。

所以最后，我想要一个数组weights = [2, 2, 2, 3, 3, 4]

在numpy中有有效的方法吗？

答案

numpy.repeat执行完全您想要的：

numpy.repeat