根据列名称获取R中的列号

Question

可能重复:
从数据框中的标签获取列索引

我需要获取给定名称的列的列号.

我们有以下数据框:

df <- data.frame(a=rnorm(100),b=rnorm(100),c=rnorm(100))

我需要一个像下面这样工作的函数:

getColumnNumber(df,"b")

它会回来

[1] 2

有这样的功能吗？

谢谢!

Answer 1

which( colnames(df)=="b" )

应该这样做.

Answer 2

一个快速而简洁的方法是:

> match("b",names(df))
[1] 2

这避免了矢量扫描是==和which做的.如果你有很多专栏,并且你做了很多,那么你可能会喜欢fastmatch包.

> require(fastmatch)
> fmatch("b",names(df))
[1] 2

fmatch比match它更快,但在随后的调用中,它不仅更快,而且是即时的.

Answer 3

另一种更好地推广到非精确匹配任务的方法是使用grep:

grep("^b$", colnames(df) )

如果您想要删除以"b"开头的所有列名的数量,您可以写:

df[ , - grep("^b", colnames(df) )]

这整齐地解决了你不能使用带有字符向量的负索引的问题.

Answer 4

..特别是,如果您需要获得多个列索引,则以下方法适用:

> df <- data.frame(a=rnorm(100),b=rnorm(100),c=rnorm(100))
> which(names(df)%in%c("b", "c"))
[1] 2 3

如果你用这个子集df你不需要哪个()

> df_sub <- df[, names(df)%in%c("b", "c")]
> head(df_sub)
           b          c
1  0.1712754  0.3119079
2 -1.3656995  0.7111664
3 -0.2176488  0.7714348
4 -0.6599826 -0.3528118
5  0.4510227 -1.6438053
6  0.2451216  2.5305453