重复指定次数的数组行

Question

对朱莉娅来说是新来的，所以这可能很容易。

我有一个n×m的数组和一个长度为n的向量，我想在向量的对应元素中重复数组的每一行的次数。例如：

mat = rand(3,6)
v = vec([2 3 1])

结果应该是6乘6的数组。我尝试了repeat功能，但是

repeat(mat, inner = v)

生成一个6×18×1数组{Float64,3}:数组，因此需要v作为重复元素的维度。在matlab中，我会使用repelem（mat，v，1），我希望julia提供类似的东西。我的实际矩阵要大得多，我需要多次调用该函数，所以这个操作需要尽可能快。

Answer 1

已经讨论过向Julia Base添加类似的内容，但目前尚未实现。您可以使用StatsBase中的inverse\rle功能实现所需。jl：

julia> row_idx = inverse_rle(axes(v, 1), v)
6-element Array{Int64,1}:
 1
 1
 2
 2
 2
 3

现在你可以写：

mat[row_idx, :]

或者

@view mat[row_idx, :]

（第二个选项创建了一个视图，如果你说你的mat很大，并且你需要多次进行这样的索引-哪个选项更快将取决于你的确切用例）。