R基于向量选择数据帧的列

时间:2012-09-07 17:42:00

标签: r plyr apply

我有一个示例数据框,如下所示。

> x=data.frame(id=1:5,c1=letters[1:5],c2=letters[13:17])
> x
  id c1 c2
1  1  a  m
2  2  b  n
3  3  c  o
4  4  d  p
5  5  e  q

我想从这个数据框中创建一个向量,它根据另一个向量为每一行选择一个不同的列。所以如果那个载体是

> vars
[1] 1 2 2 1 1 
> 

我希望x,第1列中的第1行,x,第2列中的第2行,依此类推。所以预期的输出矢量(或数据帧)就是 如果是载体

a n o d e

如果是数据框

id V1
 1  a
 2  n
 3  o
 4  d
 5  e

任何帮助,非常感谢。

1 个答案:

答案 0 :(得分:3)

您可以使用矩阵“切片”数据框:

y=data.frame(1:5,c(1,2,2,1,1))
x[2:3][as.matrix(y)]

结果:

[1] "a" "n" "o" "d" "e"

让我们通过创建一个函数

来概括它
selector=function(x)matrix(c(seq_along(x),x),ncol=2)

请注意,在开始时有一列要忽略,因此将1添加到选择向量v

v=c(1,2,2,1,1)
x[selector(v+1)]

结果

[1] "a" "n" "o" "d" "e"