Question

我有两个向量x和y，各自的长度为n和p。有没有内置的方法来创建一个np x 2矩阵，它将是

x[1] y[1]
x[1] y[2]
...
x[1] y[p]
x[2] y[1]
...
x[n] y[p]

我可以用嵌套的for循环来做到这一点，但是我正在寻找一个内置函数，如果它存在的话。

Answer 1

Julia在嵌套循环中通常非常快，所以如果它们对你正常工作，你应该可以检查性能，或者只是坚持下去。

其他选项是使用repmat（这个比使用repeat更快一点）：

[repmat(x,1,length(y))'[:] repmat(y,length(x),1)[:]]

对这两种方法做了一些快速测试：

x=rand(1000)
y=rand(1000)

function withrepeat(x,y)
    [repeat(x, inner=[size(y,1)]) repeat(y, outer=[size(x,1)])]
end

function withrepmat(x,y)
    [repmat(x,1,length(y))'[:] repmat(y,length(x),1)[:]]
end

withrepeat(x,y)
elapsed time: 0.21556302 seconds (95986112 bytes allocated)

with repmat(x,y)
elapsed time: 0.075604488 seconds (56000560 bytes allocated)

不确定为什么会有这么大的差异，我认为还有改进的余地。没有尝试过Iterators.jl包中的产品功能。

此处还有更多信息：https://groups.google.com/forum/#!topic/julia-users/dtl--SyFgwY

希望这有帮助。

尝试了几个嵌套循环，确实更快：

function withloops (x,y)
    leny=length(y)
    lenx=length(x)
    m=leny*lenx
    OUT = zeros(Float64, m,2)
    c=1
    for i = 1:lenx
        for j = 1:leny
            OUT[c,1] = x[i]
            OUT[c,2] = y[j]
            c+=1
        end
    end 
    return OUT
end

并且rand(1000)和x的{{1}}相同。

Answer 2

这是在IterTools模块中提供的（它取代了Iterators模块）。

取自https://github.com/JuliaCollections/IterTools.jl

迭代输入的笛卡尔积中的所有组合。

示例：
using IterTools
for p in product(1:3,1:2)
    @show p
end

p = (1,1)
p = (2,1)
p = (3,1)
p = (1,2)
p = (2,2)
p = (3,2)

Answer 3

我可以这样做：

julia> x = [1, 2, 3, 4];

julia> y = [9, 8, 7];

julia> [repeat(x, inner=[size(y,1)]) repeat(y, outer=[size(x,1)])]
12x2 Array{Int64,2}:
 1  9
 1  8
 1  7
 2  9
 2  8
 2  7
 3  9
 3  8
 3  7
 4  9
 4  8
 4  7

您可能还想查看Iterators.j - 特别是product功能。

Answer 4

至少在Julia 1.3中，有一个内置函数Iterators.product，因此 Iterators.product(a, b) |> collect应该可以解决问题。

朱莉娅中两个向量的笛卡尔积

4 个答案: