是否有与 NumPy 的省略号切片语法 (...) 等效的 Julia?
Is there a Julia equivalent to NumPy's ellipsis slicing syntax (...)?
在 NumPy 中,ellipsis syntax 用于
filling in a number of :
until the number of slicing specifiers matches the dimension of the array.
(意译 this answer)。
我如何在 Julia 中做到这一点?
还没有,但如果你愿意,可以自己动手。
import Base.getindex, Base.setindex!
const .. = Val{:...}
setindex!{T}(A::AbstractArray{T,1}, x, ::Type{Val{:...}}, n) = A[n] = x
setindex!{T}(A::AbstractArray{T,2}, x, ::Type{Val{:...}}, n) = A[ :, n] = x
setindex!{T}(A::AbstractArray{T,3}, x, ::Type{Val{:...}}, n) = A[ :, :, n] =x
getindex{T}(A::AbstractArray{T,1}, ::Type{Val{:...}}, n) = A[n]
getindex{T}(A::AbstractArray{T,2}, ::Type{Val{:...}}, n) = A[ :, n]
getindex{T}(A::AbstractArray{T,3}, ::Type{Val{:...}}, n) = A[ :, :, n]
那你就可以写
> rand(3,3,3)[.., 1]
3x3 Array{Float64,2}:
0.0750793 0.490528 0.273044
0.470398 0.461376 0.01372
0.311559 0.879684 0.531157
如果您想要更精细的切片,您需要generate/expand定义或使用暂存函数。
编辑:如今,参见https://github.com/ChrisRackauckas/EllipsisNotation.jl
他们要走的路是 EllipsisNotation.jl,这为语言添加了 ..
。
示例:
julia> using EllipsisNotation
julia> x = rand(1,2,3,4,5);
julia> x[..,3] == x[:,:,:,:,3]
true
julia> x[1,..] == x[1,:,:,:,:]
true
julia> x[1,1,..] == x[1,1,:,:,:]
true
(@mschauer 已经注意到他的回答(编辑),但参考文献在最后,我觉得这个问题应该得到一个干净的最新答案。)
在 NumPy 中,ellipsis syntax 用于
filling in a number of
:
until the number of slicing specifiers matches the dimension of the array.
(意译 this answer)。
我如何在 Julia 中做到这一点?
还没有,但如果你愿意,可以自己动手。
import Base.getindex, Base.setindex!
const .. = Val{:...}
setindex!{T}(A::AbstractArray{T,1}, x, ::Type{Val{:...}}, n) = A[n] = x
setindex!{T}(A::AbstractArray{T,2}, x, ::Type{Val{:...}}, n) = A[ :, n] = x
setindex!{T}(A::AbstractArray{T,3}, x, ::Type{Val{:...}}, n) = A[ :, :, n] =x
getindex{T}(A::AbstractArray{T,1}, ::Type{Val{:...}}, n) = A[n]
getindex{T}(A::AbstractArray{T,2}, ::Type{Val{:...}}, n) = A[ :, n]
getindex{T}(A::AbstractArray{T,3}, ::Type{Val{:...}}, n) = A[ :, :, n]
那你就可以写
> rand(3,3,3)[.., 1]
3x3 Array{Float64,2}:
0.0750793 0.490528 0.273044
0.470398 0.461376 0.01372
0.311559 0.879684 0.531157
如果您想要更精细的切片,您需要generate/expand定义或使用暂存函数。
编辑:如今,参见https://github.com/ChrisRackauckas/EllipsisNotation.jl
他们要走的路是 EllipsisNotation.jl,这为语言添加了 ..
。
示例:
julia> using EllipsisNotation
julia> x = rand(1,2,3,4,5);
julia> x[..,3] == x[:,:,:,:,3]
true
julia> x[1,..] == x[1,:,:,:,:]
true
julia> x[1,1,..] == x[1,1,:,:,:]
true
(@mschauer 已经注意到他的回答(编辑),但参考文献在最后,我觉得这个问题应该得到一个干净的最新答案。)