Julia中能否用负号(-)反向访问数组元素及实现方法

可以通过重载Julia数组的getindex和setindex!函数实现负索引反向访问，以下是具体实现方案：

一维数组的负索引支持 #

先导入需要重载的基础函数，再定义针对一维数组的索引处理逻辑：

import Base: getindex, setindex!

# 重载读取操作：处理负索引
function getindex(A::AbstractArray, idx::Integer)
    idx < 0 ? getindex(A, length(A) + idx + 1) : getindex(A, idx)
end

# 重载写入操作：处理负索引
function setindex!(A::AbstractArray, val, idx::Integer)
    idx < 0 ? setindex!(A, val, length(A) + idx + 1) : setindex!(A, val, idx)
end

测试示例 #

arr = [10, 20, 30, 40]
println(arr[-1])  # 输出：40（对应最后一个元素）
println(arr[-2])  # 输出：30（对应倒数第二个元素）
arr[-1] = 50
println(arr)      # 输出：[10, 20, 30, 50]

多维数组的负索引支持 #

如果需要支持矩阵这类多维数组的负索引，可以重载多参数版本的索引函数：

# 多维数组读取：逐个转换负索引
function getindex(A::AbstractArray, idxs::Integer...)
    converted_idxs = map(enumerate(idxs)) do (dim, idx)
        idx < 0 ? size(A)[dim] + idx + 1 : idx
    end
    getindex(A, converted_idxs...)
end

# 多维数组写入：逐个转换负索引
function setindex!(A::AbstractArray, val, idxs::Integer...)
    converted_idxs = map(enumerate(idxs)) do (dim, idx)
        idx < 0 ? size(A)[dim] + idx + 1 : idx
    end
    setindex!(A, val, converted_idxs...)
end

测试示例 #

mat = [1 2 3; 4 5 6; 7 8 9]
println(mat[-1, 2])  # 输出：8（最后一行第二列）
mat[-2, -1] = 0
println(mat)         # 输出：[1 2 3; 4 5 0; 7 8 9]

注意事项 #

• 全局影响范围：上述代码针对AbstractArray类型重载，会影响所有数组子类（如Vector、Matrix）。如果只想针对特定类型，可将AbstractArray替换为具体类型（比如Vector{Int}）。
• 兼容性风险：重载基础函数可能改变依赖原有索引行为的第三方库逻辑，建议仅在自己的项目环境中使用，避免在全局Julia环境中随意重载。
• 性能表现：这种索引转换仅包含简单的条件判断和算术运算，Julia的JIT编译器会进行优化，几乎不会带来性能损耗。

内容的提问来源于stack exchange，提问作者Lucas Martins Soares