如何从数据框中提取对角线元素并将其存储在变量中？

Question

我有一个简单的9元素数据框。

  A  B  C
1  8 21  1
2 40 25 32
3 10 15 49

我想提取对角线元素并将其存储在变量中。除了一次取出一个数字存储到变量之外，还有其他更简单的方法吗？

Answer 1

您可以使用diag函数来提取矩阵的对角线：

Data <- data.frame(a = c(1,2,3), b= c(11,12,13), c = c(111,112,113))
Data2 <- as.matrix(Data)
Result <- diag(Data2)
Result #Returns 1  12 113

Answer 2

在这种情况下，因为它们都是数字，所以可以使用：

df <- data.frame(a=c(4,8,10), b = c(25,24,15), c = c(1,32,49)) df

  

df      a b c   1 4 25 1   2 8 24 32   3 10 15 49

这里取对角线。 diag(as.matrix(df))

[1] 4 24 49