Это должно быть элементарно, но я не могу понять это. У меня есть фрейм данных, и я хочу создать новую переменную как матричное умножение каждой строки на заранее заданный вектор.
library(dplyr)
data <- data.frame(matrix(1:6,2))
vector <- c(1,0,1)
##NOT WORKING
data <- data %>%
mutate(multiplication = as.numeric(data[,]) %*% vector)
mm <- function(x,y){
n <- as.numeric(x)
m <- n %*% y
print(as.numeric(m))
}
##NOT WORKING
data$mm <- lapply(data[,], function(x) mm(x,vector))
Вы можете использовать apply
:
data %>% mutate(multiplication = apply(., 1, function(x) x %*% vector))
#> X1 X2 X3 multiplication
#> 1 1 3 5 6
#> 2 2 4 6 8
Один из вариантов может быть:
data %>%
rowwise() %>%
mutate(multiplication = sum(c_across(everything()) * vector))
X1 X2 X3 multiplication
<int> <int> <int> <dbl>
1 1 3 5 6
2 2 4 6 8
использовать as.matrix
library(tidyverse)
data <- data.frame(matrix(1:6,2))
vector <- c(1,0,1)
data %>%
mutate(multiplication = as.matrix(data[,]) %*% vector)
#> X1 X2 X3 multiplication
#> 1 1 3 5 6
#> 2 2 4 6 8
Created on 2020-12-11 by the reprex package (v0.3.0)
Мы можем использовать crossprod
из base R
data$multiplication <- crossprod(t(data), vector)[,1]
-выход
data
# X1 X2 X3 multiplication
#1 1 3 5 6
#2 2 4 6 8
@user20650 user20650 так что в базе R:
data$multiplication <- as.matrix(data) %*% vector