计算R

Question

我有一个像这样的数据框

df <- data.frame(id=c(1, 2, 3, 4, 5), staple_1=c("potato", "cassava","rice","fruit","coffee"), staple_2=c("cassava","beer","peanuts","rice","yams"), staple_3=c("rice","peanuts","fruit","fruit","rice"))

我也有像这样的角色矢量

staples<-c("potato","cassava","rice","yams")

我想创建一个新变量，它是“staples”字符向量中任何单词出现的行和。结果应该是这样的

df$staples<-c(3,1,1,1,2)

我尝试了几种方法，到目前为止还没有任何工作。我的实际数据帧要大得多，字符向量中有20个或更多的单词。我确信有一个简单的解决方案，但我以某种方式错过了它。

Answer 1

一个简单的apply就可以了。

apply(df, 1, function(x) sum(staples %in% x))
#[1] 3 1 1 1 2

df$staples <- apply(df, 1, function(x) sum(staples %in% x))

Answer 2

这是一个整合的解决方案。

library(tidyverse)

df <- data_frame(id = c(1, 2, 3, 4, 5), 
             staple_1 = c("potato", "cassava", "rice", "fruit", "coffee"), 
             staple_2 = c("cassava", "beer", "peanuts", "rice", "yams"), 
             staple_3 = c("rice", "peanuts", "fruit", "fruit", "rice"))

staples_vect <- c("potato", "cassava", "rice", "yams")

df %>% 
     mutate(staples = pmap_int(select(., starts_with("staple_")), ~sum(c(...) %in% staples_vect)))

使用dplyr时，最好避免使用与全局变量和数据框中的列相同的名称。

Answer 3

另一种方法是遍历列，使用%in%创建list逻辑vector并添加+）Reduce

Reduce(`+`, lapply(df[-1], `%in%`, staples))
#[1] 3 1 1 1 2

或使用rowSums将数据集转换为matrix（不包含＆＃39; id＆＃39;列），使用%in%转换为逻辑vector，指定dim转换为与df[-1]维度相同的维度并获取rowSums

rowSums(`dim<-`(as.matrix(df[-1]) %in% staples, dim(df[-1])))
#[1] 3 1 1 1 2