return the elements of an array of unknown dimension at the ith index on the first dimension

geti

geti <- function (x, i) {
  
  # return the elements of an array `x` of unknown dimension
  # at the ith index of its first dimension
  
  # throw an error if x isn't an array
  stopifnot(is.array(x))
 
  # get the dimension of x
  k <- length(dim(x))
  
  # build an expression as text
  expr <- paste('x[i,',
                paste(rep(',', k - 2), collapse = ' '),
                ']')
  
  # evaluate it
  ans <- eval(parse(text = expr))  
  
  # return this
  return (ans)

}

# # test it
# # make an array
# arr <- array(rnorm(3 * 4 * 2),
#              dim = c(3, 4, 2))
# # print it
# arr
# 
# # get the elements at 2 on the first dimension
# geti(arr, 2)

Thanks!

@BenBondLamberty pointed me towards the abind package, wherein the asub function does the trick, in the same style as your answer. So the simplest solution seems to be:

library(abind)
asub(x, i, 1)

For an array x and index i on the first dimension