在 R 编程中计算矩阵或数组的行和 - rowSums函数

有时我们需要计算矩阵或数组的行和，这时可以使用 R 语言内置函数 rowSums()。

rowSums函数的基本用法

rowSums函数的基本用法非常简单，只需要将待计算行和的矩阵或数组作为参数传入即可。以下是一个示例：

# 创建一个 3x3 的矩阵
mat <- matrix(1:9, nrow = 3)

# 计算矩阵每一行的和
rowSums(mat)

运行上述代码会输出以下结果：

[1]  6 15 24

可以看到，rowSums函数返回了一个长度为矩阵行数的向量，向量中的每一个元素都是对应行的和。

rowSums函数的高级用法

rowSums函数还支持一些高级用法，例如可以指定计算哪些行的和。以下是一些示例：

# 计算矩阵第 1 行和第 3 行的和
rowSums(mat[c(1, 3), ])

# 计算矩阵第 2 列开始的所有列的和
rowSums(mat[, 2:ncol(mat)])

# 计算一个三维数组的每个“平面”（类似于一张矩阵）的行和
arr <- array(1:18, dim = c(3, 3, 2))
apply(arr, 3, rowSums)

注意，在上述第三个示例中，我们先使用 R 语言的 apply() 函数计算每个平面的行和，然后将每个平面的结果组合成了一个三维向量。

总结

本文介绍了在 R 编程中计算矩阵或数组的行和的方法，其中重点介绍了 R 语言的内置函数 rowSums()，以及一些高级用法。rowSums函数是一个非常实用的函数，它可以帮助我们快速地计算矩阵或数组的每一行的和，提高工作效率。