在R中，根據參數，是否有任何本機函數來獲取矩陣的行或列？

Question

我目前正在使用此實現：

getVal = function(i, x, margin) {
    rst = ifelse(margin==1, x[i, ], x[, i])
}

即返回x的第i行或列，具體取決於邊距的值。

===更新===

剛意識到我在這里使用ifelse(x,y,z)語句是錯誤的，因為它返回的值與其第一個參數的長度相同。 我的getVal實現應具有：

...
rst = if (margin == 1) x[i, ] else x[, i]
...

Answer 1

abind::asub()所做的事情非常類似於您要嘗試做的事情（並且很好地推廣到了高維數組）。 它的idx和dims參數分別對應於i和margin參數。

library(abind)

(m <- matrix(1:6, ncol=2))
#      [,1] [,2]
# [1,]    1    4
# [2,]    2    5
# [3,]    3    6

asub(x = m, idx = 2, dims = 1)   # Extract 2nd row
# [1] 2 5

asub(x = m, idx = 2, dims = 2)   # Extract 2nd column
# [1] 4 5 6