R语言中mutate的使用|极客教程

R语言中mutate的使用

在R语言中，mutate是一个十分常用的函数，用于对数据框中的变量进行修改或者增加新的变量。在数据处理和分析过程中，经常需要根据已有的变量进行计算，生成新的变量，mutate函数正是用来实现这一功能的。

mutate函数的基本用法

mutate函数的基本语法如下：

mutate(.data, ...)

其中，.data是要进行操作的数据框，…表示要进行的操作，可以是对已有变量的修改，也可以是新变量的生成。下面通过一个简单的示例来说明mutate函数的基本用法。

假设我们有一个包含学生姓名和成绩的数据框，现在想要计算每位学生的总成绩，并新增一个变量用来表示。

# 创建数据框
df <- data.frame(name = c("Alice", "Bob", "Cathy"),
                 math = c(80, 85, 90),
                 english = c(75, 78, 82))

# 使用mutate计算总成绩并新增变量
library(dplyr)
df <- df %>% mutate(total = math + english)

# 查看结果
print(df)

运行结果如下：

   name math english total
1 Alice   80      75   155
2   Bob   85      78   163
3 Cathy   90      82   172

从上面的示例可以看出，通过mutate函数，我们成功地新增了一个total变量，并将每位学生的总成绩计算出来了。

mutate中的其他操作

除了简单的加减乘除运算之外，mutate函数还可以进行更加复杂的操作，如逻辑运算、条件判断、字符串处理等。下面通过一些示例来展示mutate函数的更多用法。

逻辑运算

假设我们有一个包含商品名称和价格的数据框，现在想要新增一个变量用来表示价格是否高于100元。

# 创建数据框
df <- data.frame(product = c("A", "B", "C"),
                 price = c(80, 120, 90))

# 使用mutate进行逻辑运算
df <- df %>% mutate(expensive = ifelse(price > 100, "yes", "no"))

# 查看结果
print(df)

运行结果如下：

  product price expensive
1       A    80        no
2       B   120       yes
3       C    90        no

条件判断

假设我们有一个包含学生成绩的数据框，现在想要新增一个变量用来表示成绩等级，大于90为A，80-90为B，70-80为C，70以下为D。

# 创建数据框
df <- data.frame(name = c("Alice", "Bob", "Cathy"),
                 score = c(85, 92, 78))

# 使用mutate进行条件判断
df <- df %>% mutate(grade = case_when(score > 90 ~ "A",
                                      score > 80 ~ "B",
                                      score > 70 ~ "C",
                                      TRUE ~ "D"))

# 查看结果
print(df)

运行结果如下：

   name score grade
1 Alice    85     B
2   Bob    92     A
3 Cathy    78     C

字符串处理

假设我们有一个包含学生姓名的数据框，现在想要新增一个变量用来表示姓名的首字母。

# 创建数据框
df <- data.frame(name = c("Alice", "Bob", "Cathy"))

# 使用mutate进行字符串处理
df <- df %>% mutate(first_letter = substr(name, 1, 1))

# 查看结果
print(df)

运行结果如下：

   name first_letter
1 Alice            A
2   Bob            B
3 Cathy            C

通过上面的示例，我们展示了mutate函数在逻辑运算、条件判断和字符串处理等方面的用法，说明了mutate函数的强大功能和灵活性。

总结

mutate函数是R语言中非常常用的一个函数，用来对数据框中的变量进行修改或新增变量。通过对已有变量的运算或对文本进行处理，可以快速生成新的变量，方便进行数据处理和分析。在实际的数据处理过程中，mutate函数是非常有用的工具，能够提高工作效率和数据处理的灵活性。

R语言中mutate的使用