根据具有相同长度的其他数组重复数组的元素？喜欢numpy的废除

Question

是否存在内置函数或向量有效组合，以基于数组u上的元素重复数组v上的每个元素。它类似于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])