R 函数矩阵

创建一个函数ans（a），该函数以10x10矩阵a作为输入，包含元素0和1。函数ans必须找到1的元素 并将其索引保存到一个2列矩阵中，以便第一列具有行索引，第二列具有列索引，然后返回该索引矩阵

我能找到一个解决办法，但它没有被接受，不知道出了什么问题。。有人能帮忙吗

ans <- function(a){
  a <- matrix(sample(0:1, 100, replace = T), ncol=10)
  ok <- which(a==T, arr.ind=T)
  return(ok)
}

---

ans按照说明，您必须首先创建矩阵A，然后将其传递给函数：

set.seed(505)
A <- matrix(sample(0:1, 100, replace = T), ncol=10)

ans <- function(A){
  ok <- which(A==1, arr.ind=TRUE)
  return(ok)}

ans(A)

set.seed（505）
A.