当前位置：文江博客话题详情

将函数应用于数据帧变量的子部分并将其保存到原始数据帧的 R 代码

发布于 2024-10-26 18:16:33 字数 91 浏览 2 评论 0原文

我有一个像 mtcars 这样的数据框，

我想编写一个函数来搜索以“c”开头的 mtcars 的所有变量，将它们转换为字符并将它们保存回 mtcars。

原文

分享到QQ

分享到微博

如果你对这篇内容有疑问，欢迎到本站社区发帖提问参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。

发布评论

需要登录才能够评论，你可以免费注册一个本站的账号。

原野 2024-11-02 18:16:33

这是一种方法：

f <- function(x, str = "^var") {
    want <- grep(str, names(x))
    x[, want] <- sapply(x[, want], as.character)
    x
}

[编辑：实现 @hadley 在评论中提出的观点的替代版本。]

f <- function(x, str = "^var") {
    want <- grepl(str, names(x))
    x[, want] <- lapply(x[, want], as.character)
    x
}

该函数允许您传递要匹配的字符串，但默认为 "^var" 按照您的要求。 "^" 表示我们要查找以 "var" 开头的名称，而不是任何包含 "var" 的名称。

例子：

> bigdf <- data.frame(var1 = LETTERS[1:3], var2 = letters[1:3], var3 = 1:3, 
+                     stuff = 4:6)
> bigdf
  var1 var2 var3 stuff
1    A    a    1     4
2    B    b    2     5
3    C    c    3     6
> 
> bigdf2 <- f(bigdf)
> bigdf2
  var1 var2 var3 stuff
1    A    a    1     4
2    B    b    2     5
3    C    c    3     6
> str(bigdf2)
'data.frame':   3 obs. of  4 variables:
 $ var1 : chr  "A" "B" "C"
 $ var2 : chr  "a" "b" "c"
 $ var3 : chr  "1" "2" "3"
 $ stuff: int  4 5 6

Here is one way:

f <- function(x, str = "^var") {
    want <- grep(str, names(x))
    x[, want] <- sapply(x[, want], as.character)
    x
}

[Edit: alternative version that implements the points raised by @hadley in comments.]

f <- function(x, str = "^var") {
    want <- grepl(str, names(x))
    x[, want] <- lapply(x[, want], as.character)
    x
}

The function allows you to pass through the string you want to match on, but defaults to "^var" as you requested. The "^" indicates we want to find names that start with "var", not any name that contains "var".

Example:

> bigdf <- data.frame(var1 = LETTERS[1:3], var2 = letters[1:3], var3 = 1:3, 
+                     stuff = 4:6)
> bigdf
  var1 var2 var3 stuff
1    A    a    1     4
2    B    b    2     5
3    C    c    3     6
> 
> bigdf2 <- f(bigdf)
> bigdf2
  var1 var2 var3 stuff
1    A    a    1     4
2    B    b    2     5
3    C    c    3     6
> str(bigdf2)
'data.frame':   3 obs. of  4 variables:
 $ var1 : chr  "A" "B" "C"
 $ var2 : chr  "a" "b" "c"
 $ var3 : chr  "1" "2" "3"
 $ stuff: int  4 5 6

回复收藏 0 原文

慕烟庭风 2024-11-02 18:16:33

另一种变体说明了 grep 的另一个功能和另一个版本的“[”运算符：

f <- function(x, str = "^var") {
    want <- grep(str, names(x), value=TRUE)
    x[want] <- lapply(x[want], as.character)
    x
}

value=TRUE 参数返回匹配的名称而不是它们的位置。您实际上并不需要使用 [ , Want] ，只需使用 [want] 因为这是一个完整的按列操作。检查它（和辛普森的版本）是否返回完整的 data.frame，即使没有列匹配。请不要“接受”这个答案......它只不过是一条评论，但很难阅读评论中的函数结构。

Another variation that illustrates another features of grep and another version the "[" operator:

f <- function(x, str = "^var") {
    want <- grep(str, names(x), value=TRUE)
    x[want] <- lapply(x[want], as.character)
    x
}

The value=TRUE argument returns the matching names rather than their position. You don't really need to use [ , want] and can instead just use [want]since this is a full column-wise operation. Checked to see that it (and Simpson's versions) returns the full data.frame even if no columns match. Please don't "accept" this answer ... it's not much more than a comment, but it's really hard to read function structure in the comments.

回复收藏 0 原文

~没有更多了~