小编skl*_*nbo的帖子

复制数组列

我有一个算法,需要一个数组的一列被同一个数组的另一列替换.我尝试用切片和元素方式做.

const M = 10^4
const N = 10^4
A = rand(Float32, M, N)
B = rand(Float32, N, M)

function copy_col!(A::Array{Float32,2},col1::Int,col2::Int)
  A[1:end,col2] = A[1:end,col1]
end

function copy_col2!(A::Array{Float32,2},col1::Int,col2::Int)
  for i in 1:size(A,1)
    A[i,col2] = A[i,col1]
  end
end

[Both functions+rand are called here once for compilation]

@time (for i in 1:20000 copy_col!(B, rand(1:size(B,2)),rand(1:size(B,2)) ); end )
@time (for i in 1:20000 copy_col2!(B, rand(1:size(B,2)),rand(1:size(B,2)) ); end )

>>  0.607899 seconds (314.81 k allocations: 769.879 MB, 25.05% gc time)
>>  0.213387 seconds (117.96 k …
Run Code Online (Sandbox Code Playgroud)

julia

4
推荐指数
1
解决办法
206
查看次数

标签 统计

julia ×1