Question

有没有办法覆盖[]以补充数组中的范围？

julia> a=[1:8...]
8-element Array{Int64,1}:
 1
 2
 3
 4
 5
 6
 7
 8

julia> a[-1] == a[2:8]
julia> a[-(1:3)] == a[4:8]
julia> a[-end] == a[1:7]

Answer 1

我之前没有研究过索引的内部结构，但乍看之下，以下内容可能会运行而不会破坏太多：

immutable Not{T}
    idx::T
end    

if :to_indices in names(Base)
    # 0.6
    import Base: to_indices, uncolon, tail, _maybetail

    @inline to_indices(A, inds, I::Tuple{Not, Vararg{Any}}) =
       (setdiff(uncolon(inds, (:, tail(I)...)), I[1].idx), to_indices(A, _maybetail(inds), tail(I))...)       
else
    # 0.5
    import Base: getindex, _getindex

    not_index(a::AbstractArray, I, i::Int) = I
    not_index(a::AbstractArray, I::Not, i::Int) = setdiff(indices(a, i), I.idx)

    getindex(a::AbstractArray, I::Not) = getindex(a, setdiff(linearindices(a), I.idx))
    _getindex(::Base.LinearIndexing, a::AbstractArray, I::Vararg{Union{Real, AbstractArray, Colon, Not}}) = 
        Base._getindex(Base.linearindexing(a), a, (not_index(a, idx, i) for (i,idx) in enumerate(I))...)
end

例如：

julia> a = reshape(1:9, (3, 3))
3×3 Base.ReshapedArray{Int64,2,UnitRange{Int64},Tuple{}}:
1  4  7
2  5  8
3  6  9

julia> a[Not(2:8)]
2-element Array{Int64,1}:
1
9

julia> a[Not(1:2), :]
1×3 Array{Int64,2}:
3  6  9


julia> a[Not(end), end]
2-element Array{Int64,1}:
7
8

我不关心性能，也没有进行过广泛的测试，所以事情当然可以改进。

修改

我用评论中链接的github评论替换了Matt B.版本的0.6代码。

由于他对0.6的数组索引实现的出色设计，只需要扩展一个函数来获得getindex，setindex和view的补码索引，例如< / p>
julia> view(a, Not(2:8)) 2-element SubArray{Int64,1,UnitRange{Int64},Tuple{Array{Int64,1}},false}: 1 9 # collect because ranges are immutable julia> b = collect(a); b[Not(2), Not(2)] = 10; b 3×3 Array{Int64,2}: 10 4 10 2 5 8 10 6 10

Answer 2

直接覆盖[]（即getindex）很容易破坏Base中许多与索引相关的东西，但我们可以编写一个数组包装器来解决它。我们只需要定义以下三种方法来传递您的特定测试用例：

immutable ComplementVector{T} <: AbstractArray{T,1}
    data::Vector{T}
end
Base.size(A:: ComplementVector) = size(A.data)
Base.getindex(A:: ComplementVector, i::Integer) = i > 0 ? A.data[i] : A.data[setdiff(1:end, (-i))]
Base.getindex(A:: ComplementVector, I::StepRange) = all(x->x>0, I) ? A.data[I] : A.data[setdiff(1:end, -I)]

julia> a = ComplementVector([1:8...])

julia> a[-1] == a[2:8]
true

julia> a[-(1:3)] == a[4:8]
true

julia> a[-end] == a[1:7]
true

如果您想进一步延长ComplementVector，请阅读有关Interfaces的文档。

更新

为了安全起见，我们最好不要延伸AbstractArray，因为@Fengyang Wang在评论中提出了建议：

immutable ComplementVector{T}
    data::Vector{T}
end
Base.endof(A::ComplementVector) = length(A.data)
Base.getindex(A::ComplementVector, i::Integer) = i > 0 ? A.data[i] : A.data[setdiff(1:end, (-i))]
Base.getindex(A::ComplementVector, I::OrdinalRange) = all(x->x>0, I) ? A.data[I] : A.data[setdiff(1:end, -I)]

阵列范围补充

2 个答案:

更新