一次访问数组一些元素并回绕到开头

Question

我有一个简单的问题应该有一个简单的答案，但我还没有想到它。我想一次处理一个数组，一定数量的元素，然后回绕到开头。

这是一张图表，显示n何时为10，每次都需要三个元素：

到目前为止，我尝试编写一个简单的迭代失败了：使用% n给了我零，它不能与Julia的一个索引一起工作......：）

Answer 1

提供mod1功能以允许您需要的行为：

julia> mod1(1, 5)
1

julia> mod1(3, 5)
3

julia> mod1(5, 5)
5

julia> mod1(6, 5)
1

制作mod-indexed函数非常简单：

modindex(A, i) = A[mod1(i, length(A))]

甚至你自己的mod-indexed数组类型：

julia> immutable CircArray{T} <: AbstractArray{T,1}
           xs::Vector{T}
       end

julia> Base.size(x::CircArray) = (length(x.xs),)

julia> Base.getindex(x::CircArray, i) = x.xs[mod1(i, length(x.xs))]

julia> A = CircArray([1:2:10;])
CircArray{Array{Int64,1}}([1,3,5,7,9])

julia> A[0]
9

julia> A[5]
9

julia> A[7]
3

在此基础上实现切片并不太难。正如DNF在评论中提到的那样，一个简洁明了的解决方案是

modindex(A, i) = A[mod1.(i, end)]

或getindex的等价物，它处理标量索引和切片。

编辑：由于你的问题提到了迭代，我想我提供了一个更通用的解决方案，它也适用于非数组用于迭代目的，只使用Base中的功能性迭代：

julia> threes(A) = let cy = cycle(A)
           take(zip(cy, drop(cy, 1), drop(cy, 2)), length(A))
       end
threes (generic function with 1 method)

julia> for (a, b, c) in threes([1, 2, 3, 4, 5])
           println(a, b, c)
       end
123
234
345
451
512

Answer 2

Iterators.jl会为提供简单迭代器＆＃34;一次访问数组一些元素＆＃34; ：

julia> for i in partition(1:15, 3, 1)
          @show i
       end
i = (1,2,3)
i = (2,3,4)
i = (3,4,5)
i = (4,5,6)
i = (5,6,7)
i = (6,7,8)
i = (7,8,9)
i = (8,9,10)
i = (9,10,11)
i = (10,11,12)
i = (11,12,13)
i = (12,13,14)
i = (13,14,15)

正如Feng Yang Wang所建议的那样，mod1函数将＆＃34;包裹到开头的＃34; 作业。只是组合：

julia> for i in partition(1:15, 3, 1)
          @show mod1.(collect(i), 10)
       end
mod1.(collect(i),10) = [1,2,3]
mod1.(collect(i),10) = [2,3,4]
mod1.(collect(i),10) = [3,4,5]
mod1.(collect(i),10) = [4,5,6]
mod1.(collect(i),10) = [5,6,7]
mod1.(collect(i),10) = [6,7,8]
mod1.(collect(i),10) = [7,8,9]
mod1.(collect(i),10) = [8,9,10]
mod1.(collect(i),10) = [9,10,1]
mod1.(collect(i),10) = [10,1,2]
mod1.(collect(i),10) = [1,2,3]
mod1.(collect(i),10) = [2,3,4]
mod1.(collect(i),10) = [3,4,5]

Answer 3

有关

a = [1,2,3,4,5]

DO

 repmat(a',5,2)
 5x10 Array{Int64,2}:
  1  2  3  4  5  1  2  3  4  5
  1  2  3  4  5  1  2  3  4  5
  1  2  3  4  5  1  2  3  4  5
  1  2  3  4  5  1  2  3  4  5
  1  2  3  4  5  1  2  3  4  5

然后

map(i -> [a[i,i], a[i,i+1], a[i,i+2]], 1:5)
5-element Array{Array{Int64,1},1}:
[1,2,3]
[2,3,4]
[3,4,5]
[4,5,1]
[5,1,2]

<强>加成

如果你想过度设计这个，或者内存是一个问题（repmat的内存效率不是很高）

s = size(a)[2]

l(i,v) = i + v > s ? i + v - s : i + v

map(i -> [a[i,i], a[i, l(i,1)], a[i, l(i,2)]], 1:5)

Answer 4

你可以自己定义一下吗？像：

a = repmat(collect(1:10)', 10)
sza = size(a,1) #here 10
for i in 1:sza
   toget = collect(i:i+2)
   toget[toget.>sza] -= sza
   println(a[i, toget])
end
[1,2,3]
[2,3,4]
[3,4,5]
[4,5,6]
[5,6,7]
[6,7,8]
[7,8,9]
[8,9,10]
[9,10,1]
[10,1,2]

println可能是您想要的任何地方？