如何在R语言中为数据框对角元素添加对应列的列总和？

原代码的问题 #

你写的代码逻辑有误：sum(col(.))计算的是列索引矩阵的总和，不是当前列的数值总和；而且replace的向量化处理没办法实现逐列计算并修改对应对角线元素的需求。

两种可行解决方案 #

方案一：Base R 实现

先计算每列总和，再定位对角线位置完成修改，简单直接：

# 示例数据框
df <- data.frame(a = c(1,2,3), b = c(4,5,6), c = c(7,8,9))

# 计算每列总和
col_total <- colSums(df)

# 标记对角线位置的逻辑矩阵
diag_pos <- diag(nrow(df)) == 1

# 给对角元素加上对应列的总和
df[diag_pos] <- df[diag_pos] + col_total

方案二：dplyr 管道风格实现

用across遍历每一列，结合列索引定位对角线元素：

library(dplyr)

df <- df %>%
  mutate(across(everything(), ~ {
    # 计算当前列总和
    col_sum <- sum(.x)
    # 获取当前列的索引（对应对角线的行号）
    col_index <- match(cur_column(), names(df))
    # 修改对角线位置的元素
    .x[col_index] <- .x[col_index] + col_sum
    .x
  }))

验证结果 #

以示例数据框为例，处理后对角线元素会变成：

• 第1列第1行：1 + (1+2+3) = 7
• 第2列第2行：5 + (4+5+6) = 20
• 第3列第3行：9 + (7+8+9) = 33

最终数据框：

a  b  c
1 7  4  7
2 2 20  8
3 3  6 33

内容的提问来源于stack exchange，提问作者Yvan