add stack

Author: aplavin

src/functions.jl

@@ -113,6 +113,17 @@ elseif VERSION >= v"1.1"
     end
 end
 
+@static if VERSION > v"1.9-DEV"
+    function Base.stack(A::KeyedArray; dims::Colon=:)
+        data = @invoke Base.stack(A::AbstractArray; dims)
+        if !allequal(named_axiskeys(a) for a in A)
+            throw(DimensionMismatch("stack expects uniform axiskeys for all arrays"))
+        end
+        akeys = (; named_axiskeys(first(A))..., named_axiskeys(A)...)
+        KeyedArray(data; akeys...)
+    end
+end
+
 function Base.mapslices(f, A::KeyedArray; dims)
     numerical_dims = NamedDims.dim(A, dims)
     data = mapslices(f, parent(A); dims=numerical_dims)

test/_functions.jl

@@ -46,6 +46,17 @@ A3 = wrapdims(rand(Int8, 3,4,2), r='a':'c', c=2:5, p=[10.0, 20.0])
     if VERSION >= v"1.1"
         @test axiskeys(first(eachslice(M, dims=:r))) === (2:5,)
     end
+    if VERSION >= v"1.9-DEV"
+        arr = KeyedArray([
+            KeyedArray([1, 2], a=[:x, :y]),
+            KeyedArray([3, 4], a=[:x, :y]),
+            KeyedArray([5, 6], a=[:x, :y]),
+        ], b=10:12)
+
+        sk = Base.stack(arr)::KeyedArray
+        @test sk == [1 3 5; 2 4 6]
+        @test named_axiskeys(sk) == (a=[:x, :y], b=10:12)
+    end
 
     @test axiskeys(selectdim(M, :r, [true, false, true])) == (['a', 'c'], 2:5)