根据具有相同长度的其他数组重复数组的元素?就像对numpy的排斥

Question

是否有内置函数或矢量化的有效组合，可以在数组v上的元素的基础上重复数组u上的每个元素。它将类似于Matlab的repelem（具有阵列输入）。

示例：

a = np.array([1, 5, 2])
b = np.array([2, 3, 4]) # be must have same length of a
print np.repelem(a, b)

[1,1,5,5,5,2,2,2,2]

Answer 1

repeat(a, repeats)应该完成的工作

In [21]: np.repeat(np.array([1,5,2]), np.array([2,3,4]))
Out[21]: array([1, 1, 5, 5, 5, 2, 2, 2, 2])