высчитать сумму строки и продукт в data.frame

Question

Я хотел бы, чтобы присоединять колонки к моему data.frame в R, которые содержат строки суммы и продукты Рассмотрим следующий кадр данных

x y  z 
1 2  3 
2 3  4 
5 1  2 

Я хочу, чтобы получить следующее

x y  z sum prod 
1 2  3 6  6 
2 3  4 9  24 
5 1  2 8  10 

Я попытался

sum = apply(ages,1,add) 

, но это дает мне вектор-строку. Может кто-нибудь, пожалуйста, покажет мне эффективную команду суммировать и продукт и добавить их в исходный фрейм данных, как показано выше?

Answer 1

Попробуйте

transform(df, sum=rowSums(df), prod=x*y*z) 
# x y z sum prod 
#1 1 2 3 6 6 
#2 2 3 4 9 24 
#3 5 1 2 8 10 

Или

transform(df, sum=rowSums(df), prod=Reduce(`*`, df)) 
# x y z sum prod 
#1 1 2 3 6 6 
#2 2 3 4 9 24 
#3 5 1 2 8 10 

Другим вариантом было бы использовать rowProdsmatrixStats из

library(matrixStats) 
transform(df, sum=rowSums(df), prod=rowProds(as.matrix(df))) 

Если вы используете apply

df[,c('sum', 'prod')] <- t(apply(df, 1, FUN=function(x) c(sum(x), prod(x)))) 
df 
# x y z sum prod 
#1 1 2 3 6 6 
#2 2 3 4 9 24 
#3 5 1 2 8 10 

Answer 2

Другой подход.

require(data.table) 

# Create data 
dt <- data.table(x = c(1,2,5), y = c(2,3,1), z = c(3,4,2)) 

# Create index 
dt[, i := .I] 

# Compute sum and prod 
dt[, sum := sum(x, y, z), by = i] 
dt[, prod := prod(x, y, z), by = i] 
dt 


# Compute sum and prod using .SD 
dt[, c("sum", "prod") := NULL] 
dt 
dt[, sum := sum(.SD), by = i, .SDcols = c("x", "y", "z")] 
dt[, prod := prod(.SD), by = i, .SDcols = c("x", "y", "z")] 
dt 


# Compute sum and prod using .SD and list 
dt[, c("sum", "prod") := NULL] 
dt 
dt[, c("sum", "prod") := list(sum(.SD), prod(.SD)), by = i, 
    .SDcols = c("x", "y", "z")] 
dt 


# Compute sum and prod using .SD and lapply 
dt[, c("sum", "prod") := NULL] 
dt 
dt[, c("sum", "prod") := lapply(list(sum, prod), do.call, .SD), by = i, 
    .SDcols = c("x", "y", "z")] 
dt 

Answer 3

После также может быть сделано, но имена столбцов должны быть введены:

ddf$sum = with(ddf, x+y+z) 
ddf$prod = with(ddf, x*y*z) 
ddf 
    x y z sum prod 
1 1 2 3 6 6 
2 2 3 4 9 24 
3 5 1 2 8 10 

С data.table, другая форма может быть:

library(data.table)  
cbind(dt, dt[,list(sum=x+y+z, product=x*y*z),]) 
    x y z sum product 
1: 1 2 3 6  6 
2: 2 3 4 9  24 
3: 5 1 2 8  10 

Более простой вариант предложен @David Arenberg в комментариях:

dt[, ":="(sum = x+y+z, product = x*y*z)]