迹忆客 专注技术分享

当前位置:主页 > 学无止境 > 编程语言 >

R 数据框的选定列的总和

作者:迹忆客 最近更新:2023/03/21 浏览次数:

在很多情况下,我们想使用其他列的值创建一个新列。本文将教授如何创建一个新列来计算 R 中所选数据框列的总和。

我们将学习创建新列的三种方法:使用基础 R 中的 rowSums()apply() 以及 Tidyverse 中的一组函数。

我们可以根据使用的函数以不同的方式指定要添加的列:直接在函数中、使用字符串向量或模式匹配函数。

我们将首先创建一个数据框。该数据框没有行名,也没有缺失值。

示例代码:

# Create five variables.
Student = c("Student A", "Student B", "Student C")
Hobby = c("Music", "Sports", "Cycling")
Maths = c(40, 35, 30)
Statistics = c(30, 35, 20)
Programming = c(25, 20, 35)

# Create a data frame from the variables.
df_students = data.frame(Student, Hobby, Maths, Statistics, Programming)

# View the data frame.
df_students

使用 Base R 的 rowSums() 函数计算数据框选定列的总和

我们将使用 data_frame$new_column 语法创建一个新列,并使用 rowSums() 函数为其赋值。要添加的列将使用子集语法直接在函数中给出。

示例代码:

# This adds the new column to the data frame.
df_students$myRowSums = rowSums(df_students[,c("Maths", "Statistics", "Programming")])

# View the data frame with the added column.
df_students

# We can also give a vector of column positions.
# df_students$myRowSums = rowSums(df_students[,c(3:5)])

输出:

> # View the data frame with the added column.
> df_students
    Student   Hobby Maths Statistics Programming myRowSums
1 Student A   Music    40         30          25        95
2 Student B  Sports    35         35          20        90
3 Student C Cycling    30         20          35        85

我们还可以保存列的名称以添加为字符串向量。我们可以将此向量传递给 rowSums() 函数。

示例代码:

# Save the list of columns as a vector of strings.
col_list = c("Maths", "Statistics", "Programming")

# Pass the vector of strings to the subsetting square brackets.
df_students$myRowSums = rowSums(df_students[,col_list])

# View the data frame with the added column.
df_students

文档指出,rowSums() 函数等效于带有 FUN = sumapply() 函数,但要快得多。

它指出 rowSums() 函数模糊了一些 NaNNA 的细微之处。

使用 Base R 的 apply() 函数计算数据框选定列的总和

我们将这三个参数传递给 apply() 函数。

  1. 数据框所需的列。
  2. 要保留的数据框的维度。1 表示行。
  3. 我们要计算的函数,sum

示例代码:

# We will recreate the data frame from the variables.
df_students = data.frame(Student, Hobby, Maths, Statistics, Programming)

# In base R, we can delete a column by setting its name to NULL.
# df_students$myRowSums = NULL

# A new column gets created.
df_students$myApplySums = apply(df_students[,col_list], 1, sum)

# View the data frame with the added column.
df_students

我们还可以将要添加的列作为字符串或位置的向量直接添加到方括号中。

示例代码:

# Names of columns as a vector of strings.
df_students$myApplySums = apply(df_students[,c("Maths", "Statistics", "Programming")], 1, sum)

# Vector of columns positions.
df_students$myApplySums = apply(df_students[,c(3, 4, 5)], 1, sum)

使用 Tidyverse 函数计算 R 中数据框选定列的总和

我们可以将 dplyrmutate() 函数与 Tidyverse 中的其他函数结合使用来创建总和列。

在使用 Tidyverse 方法时,我们需要了解一些细节。Tibbles 删除行名称,并且对有效数字、尾随零和尾随小数具有不同的默认值。

首先,我们需要加载 dplyr 包并创建一个 tibble。

请注意我们如何在代码中使用以下内容。

  1. 管道运算符,%>%,以避免嵌套某些函数。
  2. rowwise() 使其他函数在行上工作。
  3. mutate() 添加列。
  4. sum() 用于加法。
  5. c_across() 旨在与 rowwise() 一起使用。
  6. all_of() 从字符向量中选择值。

rowwise() 是一种分组类型。使用后,我们可能需要使用 ungroup(data_frame_name) 并将未分组的版本保存为对象。

示例代码:

# We will recreate the data frame from the variables.
df_students = data.frame(Student, Hobby, Maths, Statistics, Programming)

# Load the dplyr package.
library(dplyr)

# Create a tibble from the data frame.
# This could have been done with the next step but obscured the main point.
tb_students = as_tibble(df_students)


# We have to assign the RHS to an object to save the column to the object.
# It can be the same as the original tibble.
tb_students = tb_students %>% rowwise() %>% mutate(myTidySum = sum(c_across(all_of(col_list))))

# View the rowwise tibble with the added column.
tb_students

输出:

> # View the rowwise tibble with the added column.
> tb_students
# A tibble: 3 x 6
# Rowwise:
  Student   Hobby   Maths Statistics Programming myTidySum
  <chr>     <chr>   <dbl>      <dbl>       <dbl>     <dbl>
1 Student A Music      40         30          25        95
2 Student B Sports     35         35          20        90
3 Student C Cycling    30         20          35        85

上面的示例使用了四个函数。它比前几节的方法更复杂。

但是 Tidyverse 方法的一大优势是它提供了许多指定列的方法。

根据我们的需要,我们可以使用选择辅助函数,例如 starts_with()contains()where(),使用联合和交集组合选择,指定我们不想选择的列,等等在。

下面的示例代码提供了五种等效方法来实现相同的结果。由于这种灵活性,我们需要熟悉这种方法。

示例代码:

tb_students = as_tibble(df_students)
# Take the union of the column names.
tb_students = tb_students %>% rowwise() %>% mutate(myTidySum = sum(c_across(Maths | Statistics | Programming)))


tb_students = as_tibble(df_students)
# Give a range of columns as a range of names.
tb_students = tb_students %>% rowwise() %>% mutate(myTidySum = sum(c_across(Maths:Programming)))


tb_students = as_tibble(df_students)
# Give a range of columns as a range of column positions.
tb_students = tb_students %>% rowwise() %>% mutate(myTidySum = sum(c_across(3:5)))


tb_students = as_tibble(df_students)
# Select all columns having 'at' or 'am'
tb_students = tb_students %>% rowwise() %>% mutate(myTidySum = sum(c_across(contains('at') | contains('am'))))


tb_students = as_tibble(df_students)
# Select all columns except Student and Hobby.
# Make sure the tibble only has the required columns before running the next line.
tb_students = tb_students %>% rowwise() %>% mutate(myTidySum = sum(c_across(!c(Student, Hobby))))

参考和帮助

在 R Studio 中,有关 rowSums()apply() 的帮助,请单击 Help > Search R Help 并在搜索框中键入不带括号的函数名称。或者,在 R 控制台的命令提示符处键入一个问号,后跟函数名称。

结论

rowSums()apply() 函数使用简单。要添加的列可以使用名称或列位置直接在函数中指定,也可以作为字符向量提供。

Tidyverse 方法虽然有点复杂,但提供了许多替代方法来指定要添加的列。

我们可以选择最适合我们需求的方法。

转载请发邮件至 1244347461@qq.com 进行申请,经作者同意之后,转载请以链接形式注明出处

本文地址:

相关文章

R 中具有多个条件的函数向量化

发布时间:2023/03/21 浏览次数:64 分类:编程语言

一项常见的数据分析任务是根据同一行的其他列使用一个或多个条件创建或更新数据框列。 如果我们尝试使用 if 语句来执行此操作,则只会使用第一行来测试条件,并且会根据该行更

在 R 中读取 xlsx 文件

发布时间:2023/03/21 浏览次数:66 分类:编程语言

在这篇文章中,你将会了解到两个在 R 中读取 xlsx 文件的最完整和最容易使用的库:readxl 和 openxlsx。

清理 R 的环境

发布时间:2023/03/21 浏览次数:178 分类:编程语言

在本教程中,你将学习如何在 R 中编写一个函数,在不需要重新启动 R 的情况下清除环境。

在 R 中注释掉多行

发布时间:2023/03/21 浏览次数:63 分类:编程语言

在本文中,你将学习如何在 R 中注释出多行,而不必在每一行的开头手动写一个#字符来注释。

在 R 中清除内存

发布时间:2023/03/21 浏览次数:197 分类:编程语言

在本教程中,你将学习如何清除 R 系统占用的内存,而不必重新启动它或重新启动它运行的计算机。

扫一扫阅读全部技术教程

社交账号
  • https://www.github.com/onmpw
  • qq:1244347461

最新推荐

教程更新

热门标签

扫码一下
查看教程更方便